fix

zhyncs · zhyncs · commit 4d23df0a07b0 · 2025-10-03T17:28:20.000-07:00
diff --git a/csrc/jit_kernels/impls/sm90_fp8_gemm_1d1d.hpp b/csrc/jit_kernels/impls/sm90_fp8_gemm_1d1d.hpp
@@ -131,7 +131,7 @@ static void sm90_fp8_gemm_1d1d(const torch::Tensor& a, const torch::Tensor& sfa,
     const auto& code = SM90FP8Gemm1D1DRuntime::generate(args);
     const auto& runtime = compiler->build("sm90_fp8_gemm_1d1d", code);
 
-    SM90FP8Gemm1D1DRuntime::launch(runtime, args);
+    MAYBE_LAUNCH(SM90FP8Gemm1D1DRuntime::launch(runtime, args));
 }
 
 static void sm90_fp8_k_grouped_gemm_1d1d(const torch::Tensor& a, const torch::Tensor& sfa,
@@ -206,7 +206,7 @@ static void sm90_fp8_k_grouped_gemm_1d1d(const torch::Tensor& a, const torch::Te
     const auto& code = SM90FP8Gemm1D1DRuntime::generate(args);
     const auto& runtime = compiler->build("sm90_fp8_gemm_1d1d", code);
 
-    SM90FP8Gemm1D1DRuntime::launch(runtime, args);
+    MAYBE_LAUNCH(SM90FP8Gemm1D1DRuntime::launch(runtime, args));
 }
 
 } // namespace deep_gemm

Original file line number	Diff line number	Diff line change
`@@ -131,7 +131,7 @@ static void sm90_fp8_gemm_1d1d(const torch::Tensor& a, const torch::Tensor& sfa,`
`131`	`131`	`const auto& code = SM90FP8Gemm1D1DRuntime::generate(args);`
`132`	`132`	`const auto& runtime = compiler->build("sm90_fp8_gemm_1d1d", code);`
`133`	`133`
`134`		`- SM90FP8Gemm1D1DRuntime::launch(runtime, args);`
	`134`	`+ MAYBE_LAUNCH(SM90FP8Gemm1D1DRuntime::launch(runtime, args));`
`135`	`135`	`}`
`136`	`136`
`137`	`137`	`static void sm90_fp8_k_grouped_gemm_1d1d(const torch::Tensor& a, const torch::Tensor& sfa,`
`@@ -206,7 +206,7 @@ static void sm90_fp8_k_grouped_gemm_1d1d(const torch::Tensor& a, const torch::Te`
`206`	`206`	`const auto& code = SM90FP8Gemm1D1DRuntime::generate(args);`
`207`	`207`	`const auto& runtime = compiler->build("sm90_fp8_gemm_1d1d", code);`
`208`	`208`
`209`		`- SM90FP8Gemm1D1DRuntime::launch(runtime, args);`
	`209`	`+ MAYBE_LAUNCH(SM90FP8Gemm1D1DRuntime::launch(runtime, args));`
`210`	`210`	`}`
`211`	`211`
`212`	`212`	`} // namespace deep_gemm`