Import Tint changes from Dawn
Changes:
- 635977a94ad7731ff6de419318235b2c6fa0a243 [tint] Generate ArrayLengthFromUniform indices by James Price <jrprice@google.com>
GitOrigin-RevId: 635977a94ad7731ff6de419318235b2c6fa0a243
Change-Id: I251e2a32f481a5c0f1b88c1c4c1c004b97bdb55c
Reviewed-on: https://dawn-review.googlesource.com/c/tint/+/163705
Kokoro: Kokoro <noreply+kokoro@google.com>
Commit-Queue: Ben Clayton <bclayton@google.com>
Reviewed-by: Ben Clayton <bclayton@google.com>
diff --git a/src/tint/cmd/tint/main.cc b/src/tint/cmd/tint/main.cc
index 5ba2c6e..07bf73b 100644
--- a/src/tint/cmd/tint/main.cc
+++ b/src/tint/cmd/tint/main.cc
@@ -33,6 +33,7 @@
#include <string>
#include <unordered_map>
#include <vector>
+#include "src/tint/lang/wgsl/sem/variable.h"
#if TINT_BUILD_SPV_READER || TINT_BUILD_SPV_WRITER
#include "spirv-tools/libspirv.hpp"
@@ -763,10 +764,19 @@
gen_options.pixel_local_options = options.pixel_local_options;
gen_options.bindings = tint::msl::writer::GenerateBindings(*input_program);
gen_options.array_length_from_uniform.ubo_binding = tint::BindingPoint{0, 30};
- gen_options.array_length_from_uniform.bindpoint_to_size_index.emplace(tint::BindingPoint{0, 0},
- 0);
- gen_options.array_length_from_uniform.bindpoint_to_size_index.emplace(tint::BindingPoint{0, 1},
- 1);
+
+ // Add array_length_from_uniform entries for all storage buffers with runtime sized arrays.
+ std::unordered_set<tint::BindingPoint> storage_bindings;
+ for (auto* var : program.AST().GlobalVariables()) {
+ auto* sem_var = program.Sem().Get<tint::sem::GlobalVariable>(var);
+ if (!sem_var->Type()->UnwrapRef()->HasFixedFootprint()) {
+ auto bp = sem_var->Attributes().binding_point.value();
+ if (storage_bindings.insert(bp).second) {
+ gen_options.array_length_from_uniform.bindpoint_to_size_index.emplace(
+ bp, static_cast<uint32_t>(storage_bindings.size() - 1));
+ }
+ }
+ }
tint::Result<tint::msl::writer::Output> result;
if (options.use_ir) {
