Bugfix.

huggingface · May 25, 2024 · 093792f · 093792f
1 parent a73019e
commit 093792f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/candle-core/src/metal_backend/mod.rs b/candle-core/src/metal_backend/mod.rs
@@ -873,7 +873,7 @@ impl BackendStorage for MetalStorage {
                 &command_buffer,
                 &self.device.kernels,
                 name,
-                layout.dims(),
+                &[b_size, l_in, c_out, k_size],
                 params.k_size,
                 params.stride,
                 BufferOffset::zero_offset(&col.buffer),