mirror of https://github.com/microsoft/vcpkg.git
15 lines
753 B
Diff
15 lines
753 B
Diff
--- a/aten/src/ATen/CMakeLists.txt
|
|
+++ b/aten/src/ATen/CMakeLists.txt
|
|
@@ -797,8 +797,9 @@
|
|
add_definitions(-DCUTLASS_ENABLE_TENSOR_CORE_MMA=1)
|
|
add_definitions(-DCUTLASS_ENABLE_SM90_EXTENDED_MMA_SHAPES=1)
|
|
add_definitions(-DCUTE_SM90_EXTENDED_MMA_SHAPES_ENABLED)
|
|
- list(APPEND ATen_CUDA_INCLUDE ${CMAKE_CURRENT_SOURCE_DIR}/../../../third_party/cutlass/include)
|
|
- list(APPEND ATen_CUDA_INCLUDE ${CMAKE_CURRENT_SOURCE_DIR}/../../../third_party/cutlass/tools/util/include)
|
|
+ find_package(NvidiaCutlass CONFIG REQUIRED)
|
|
+ set_target_properties(nvidia::cutlass::cutlass PROPERTIES IMPORTED_GLOBAL TRUE)
|
|
+ list(APPEND ATen_CUDA_DEPENDENCY_LIBS nvidia::cutlass::cutlass)
|
|
|
|
if($ENV{ATEN_STATIC_CUDA})
|
|
if(CUDA_VERSION VERSION_LESS_EQUAL 12.9)
|