add fusion matmul i64 optimization

tzemanovic · tzemanovic · commit c83418c2e932 · 2025-10-29T18:36:35.000Z
diff --git a/crates/burn-cubecl-fusion/src/matmul/optimization.rs b/crates/burn-cubecl-fusion/src/matmul/optimization.rs
@@ -369,6 +369,7 @@ impl<R: Runtime> TraceRunner<R> for FusedMatmul {
             FusePrecision::BF16 => {
                 self.matmul_fused::<R, bf16>(client, inputs, outputs, &configs[0])
             }
+            FusePrecision::I64 => self.matmul_fused::<R, i64>(client, inputs, outputs, &configs[0]),
             _ => panic!("Unsupported precision"),
         }
     }

Original file line number	Diff line number	Diff line change
`@@ -369,6 +369,7 @@ impl<R: Runtime> TraceRunner<R> for FusedMatmul {`
`369`	`369`	`FusePrecision::BF16 => {`
`370`	`370`	`self.matmul_fused::<R, bf16>(client, inputs, outputs, &configs[0])`
`371`	`371`	`}`
	`372`	`+ FusePrecision::I64 => self.matmul_fused::<R, i64>(client, inputs, outputs, &configs[0]),`
`372`	`373`	`_ => panic!("Unsupported precision"),`
`373`	`374`	`}`
`374`	`375`	`}`