mirror of
https://github.com/encounter/dawn-cmake.git
synced 2025-07-25 06:25:42 +00:00
e667d777ac
Should reduce memory and improve performance. Bug: chromium:1360731 Bug: chromium:1357188 Change-Id: I94c3653f6d61db9d949ad27d8b666688ccad86dd Reviewed-on: https://dawn-review.googlesource.com/c/dawn/+/102182 Reviewed-by: Antonio Maiorano <amaiorano@google.com> Commit-Queue: Antonio Maiorano <amaiorano@google.com> Kokoro: Kokoro <noreply+kokoro@google.com> Auto-Submit: Ben Clayton <bclayton@google.com> Commit-Queue: Ben Clayton <bclayton@google.com>
128 lines
4.0 KiB
C++
128 lines
4.0 KiB
C++
// Copyright 2022 The Tint Authors.
|
|
//
|
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
|
// you may not use this file except in compliance with the License.
|
|
// You may obtain a copy of the License at
|
|
//
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
|
//
|
|
// Unless required by applicable law or agreed to in writing, software
|
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
// See the License for the specific language governing permissions and
|
|
// limitations under the License.
|
|
|
|
#include <fstream>
|
|
#include <iostream>
|
|
#include <string>
|
|
#include <unordered_set>
|
|
|
|
#include <thread>
|
|
|
|
#include "src/tint/inspector/inspector.h"
|
|
#include "src/tint/reader/wgsl/parser.h"
|
|
#include "src/tint/utils/hash.h"
|
|
#include "src/tint/writer/flatten_bindings.h"
|
|
#include "src/tint/writer/glsl/generator.h"
|
|
#include "src/tint/writer/hlsl/generator.h"
|
|
#include "src/tint/writer/msl/generator.h"
|
|
#include "src/tint/writer/spirv/generator.h"
|
|
#include "src/tint/writer/wgsl/generator.h"
|
|
|
|
static constexpr size_t kNumThreads = 8;
|
|
|
|
[[noreturn]] void TintInternalCompilerErrorReporter(const tint::diag::List& diagnostics) {
|
|
auto printer = tint::diag::Printer::create(stderr, true);
|
|
tint::diag::Formatter{}.format(diagnostics, printer.get());
|
|
__builtin_trap();
|
|
}
|
|
|
|
extern "C" int LLVMFuzzerTestOneInput(const uint8_t* data, size_t size) {
|
|
tint::SetInternalCompilerErrorReporter(&TintInternalCompilerErrorReporter);
|
|
|
|
std::string str(reinterpret_cast<const char*>(data), size);
|
|
auto file = std::make_unique<tint::Source::File>("test.wgsl", str);
|
|
auto program = tint::reader::wgsl::Parse(file.get());
|
|
if (!program.IsValid()) {
|
|
return 0;
|
|
}
|
|
|
|
tint::inspector::Inspector inspector(&program);
|
|
auto entry_points = inspector.GetEntryPoints();
|
|
std::string entry_point = entry_points.empty() ? "" : entry_points.front().name;
|
|
|
|
std::array<std::thread, kNumThreads> threads;
|
|
|
|
for (size_t thread_idx = 0; thread_idx < kNumThreads; thread_idx++) {
|
|
auto thread = std::thread([&program, thread_idx, entry_point] {
|
|
enum class Writer {
|
|
#if TINT_BUILD_GLSL_WRITER
|
|
kGLSL,
|
|
#endif
|
|
#if TINT_BUILD_HLSL_WRITER
|
|
kHLSL,
|
|
#endif
|
|
#if TINT_BUILD_MSL_WRITER
|
|
kMSL,
|
|
#endif
|
|
#if TINT_BUILD_SPV_WRITER
|
|
kSPIRV,
|
|
#endif
|
|
#if TINT_BUILD_WGSL_WRITER
|
|
kWGSL,
|
|
#endif
|
|
kCount
|
|
};
|
|
switch (static_cast<Writer>(thread_idx % static_cast<size_t>(Writer::kCount))) {
|
|
#if TINT_BUILD_WGSL_WRITER
|
|
case Writer::kWGSL: {
|
|
tint::writer::wgsl::Generate(&program, {});
|
|
break;
|
|
}
|
|
#endif // TINT_BUILD_WGSL_WRITER
|
|
|
|
#if TINT_BUILD_SPV_WRITER
|
|
case Writer::kSPIRV: {
|
|
tint::writer::spirv::Generate(&program, {});
|
|
break;
|
|
}
|
|
#endif // TINT_BUILD_SPV_WRITER
|
|
|
|
#if TINT_BUILD_HLSL_WRITER
|
|
case Writer::kHLSL: {
|
|
tint::writer::hlsl::Generate(&program, {});
|
|
break;
|
|
}
|
|
#endif // TINT_BUILD_HLSL_WRITER
|
|
|
|
#if TINT_BUILD_GLSL_WRITER
|
|
case Writer::kGLSL: {
|
|
tint::writer::glsl::Generate(&program, {}, entry_point);
|
|
break;
|
|
}
|
|
#endif // TINT_BUILD_GLSL_WRITER
|
|
|
|
#if TINT_BUILD_MSL_WRITER
|
|
case Writer::kMSL: {
|
|
// Remap resource numbers to a flat namespace.
|
|
if (auto flattened = tint::writer::FlattenBindings(&program)) {
|
|
tint::writer::msl::Generate(&flattened.value(), {});
|
|
}
|
|
break;
|
|
}
|
|
#endif // TINT_BUILD_MSL_WRITER
|
|
|
|
case Writer::kCount:
|
|
break;
|
|
}
|
|
});
|
|
threads[thread_idx] = std::move(thread);
|
|
}
|
|
|
|
for (auto& thread : threads) {
|
|
thread.join();
|
|
}
|
|
|
|
return 0;
|
|
}
|