Nit

axinging · Dec 3, 2024 · 07a502f · 07a502f
1 parent 26b78b4
commit 07a502f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/js/web/lib/wasm/jsep/webgpu/ops/gather-nd.ts b/js/web/lib/wasm/jsep/webgpu/ops/gather-nd.ts
@@ -39,7 +39,7 @@ const computeSliceOffsets = (
 
   const getShaderSource = (shaderHelper: ShaderHelper) => {
     const indices = inputVariable('indices_data', indicesData.dataType, indicesData.dims.length);
-    const output = outputVariable('input_slice_offsets_data', DataType.uint32, 1, 1);
+    const output = outputVariable('input_slice_offsets_data', DataType.int64, 1, 1);
     const variables = [indices, output];
     const uniforms: UniformsArrayType = [
       { name: 'output_size', type: 'u32' },