feat: support misc kernel launch

2025-10-22 06:23:08 +00:00
parent 301cbc1d75
commit f4adba8a66
9 changed files with 52 additions and 24 deletions
@@ -134,7 +134,7 @@ static void sm100_bf16_gemm(const torch::Tensor& a,
    };
    const auto& code = SM100BF16GemmRuntime::generate(args);
    const auto& runtime = compiler->build("sm100_bf16_gemm", code);
-    SM100BF16GemmRuntime::launch(runtime, args);
+    MAYBE_LAUNCH(SM100BF16GemmRuntime::launch(runtime, args));
 }

 } // namespace deep_gemm