[Bugfix] Fix DSV3 kernels breaking _C and _moe_C on unsupported arches (#35123)

Signed-off-by: mgoin <mgoin64@gmail.com>
This commit is contained in:
Michael Goin
2026-02-23 20:11:27 -05:00
committed by GitHub
parent 22a97e6613
commit 3ef9fd0f98
5 changed files with 12 additions and 3 deletions

View File

@@ -127,7 +127,7 @@ TORCH_LIBRARY_EXPAND(TORCH_EXTENSION_NAME, m) {
// DeepSeek V3 optimized router GEMM for SM90+
m.def("dsv3_router_gemm(Tensor! output, Tensor mat_a, Tensor mat_b) -> ()");
m.impl("dsv3_router_gemm", torch::kCUDA, &dsv3_router_gemm);
// conditionally compiled so impl registration is in source file
#endif
}