[BugFix] Fix: ImportError when building on hopper systems (#20513)

Signed-off-by: Lucas Wilkinson <lwilkins@redhat.com>
This commit is contained in:
Lucas Wilkinson
2025-07-06 00:17:30 -04:00
committed by GitHub
parent 432870829d
commit 40b86aa05e
4 changed files with 10 additions and 9 deletions

View File

@@ -1,3 +1,5 @@
#include "core/registration.h"
#include <torch/all.h>
#include <cutlass/arch/arch.h>
@@ -364,4 +366,9 @@ void cutlass_blockwise_scaled_grouped_mm(
TORCH_CHECK(false, "Unsupported output tensor type");
}
#endif
}
}
TORCH_LIBRARY_IMPL_EXPAND(TORCH_EXTENSION_NAME, CUDA, m) {
m.impl("cutlass_blockwise_scaled_grouped_mm",
&cutlass_blockwise_scaled_grouped_mm);
}