Skip to content

Commit b9013ac

Browse files
bottlerxFormers Bot
authored andcommitted
export cuda 12.0 compat
ghstack-source-id: dee688b628652050ce7bdc864eb4861c755e53cd Pull Request resolved: fairinternal/xformers#1377 __original_commit__ = fairinternal/xformers@0c96c00
1 parent 0bd98f5 commit b9013ac

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

xformers/csrc/sparse24/sparse24_gemm_sm90.cu

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -24,6 +24,9 @@
2424

2525
// CUTLASS does not mix well with Windows
2626
#ifndef _WIN32
27+
#if __CUDACC_VER_MAJOR__ > 12 || \
28+
(__CUDACC_VER_MAJOR__ == 12 && __CUDACC_VER_MINOR__ > 0)
29+
2730
#ifdef CUTLASS_SM90_COLLECTIVE_BUILDER_SUPPORTED
2831
namespace {
2932
#define CUTLASS_STATUS_CHECK(status) \
@@ -447,3 +450,4 @@ TORCH_LIBRARY_IMPL(xformers, Meta, m) {
447450
}
448451
#endif
449452
#endif
453+
#endif

0 commit comments

Comments
 (0)