set workspace size

LopezCastroRoberto · LopezCastroRoberto · commit 34db819b09f0 · 2025-07-14T14:01:25.000+02:00
diff --git a/qutlass/csrc/fused_quantize_mx.cu b/qutlass/csrc/fused_quantize_mx.cu
@@ -97,9 +97,9 @@ struct GemmRunner {
     const at::cuda::OptionalCUDAGuard device_guard(device_of(x));
     cudaStream_t stream = at::cuda::getCurrentCUDAStream(device.index());
 
-    gemmOp.initialize(arguments, stream);
+    gemmOp.initialize(arguments, nullptr, stream);
 
-    CUTLASS_CHECK(gemmOp(arguments, stream));
+    CUTLASS_CHECK(gemmOp(arguments, nullptr, stream));
 
     return true;
   }

Original file line number	Diff line number	Diff line change
`@@ -97,9 +97,9 @@ struct GemmRunner {`
`97`	`97`	`const at::cuda::OptionalCUDAGuard device_guard(device_of(x));`
`98`	`98`	`cudaStream_t stream = at::cuda::getCurrentCUDAStream(device.index());`
`99`	`99`
`100`		`- gemmOp.initialize(arguments, stream);`
	`100`	`+ gemmOp.initialize(arguments, nullptr, stream);`
`101`	`101`
`102`		`- CUTLASS_CHECK(gemmOp(arguments, stream));`
	`102`	`+ CUTLASS_CHECK(gemmOp(arguments, nullptr, stream));`
`103`	`103`
`104`	`104`	`return true;`
`105`	`105`	`}`