forgot to add few lines to CMakeLists for SP2TC flag

lanl · May 18, 2022 · e5ff9df · e5ff9df
1 parent c86ab0e
commit e5ff9df
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/CMakeLists.txt b/CMakeLists.txt
@@ -128,6 +128,12 @@ else()
     add_definitions(-DMETIS_REAL_KIND=0d0)
 endif()
 
+if(NOT PROGRESS_SP2TC MATCHES "no")
+    message(STATUS "Using sp2 tensorcore (PROGRESS_SP2TC): ${PROGRESS_SP2TC}") 
+    add_definitions(-DSP2TC${PROGRESS_SP2TC})
+    add_subdirectory(src/sp2_tensorcore)
+endif()
+
 set(BML_OPENMP TRUE CACHE BOOL "Whether the bml was built with OpenMP")
 set(PROGRESS_OPENMP TRUE CACHE BOOL "Whether to compile with OpenMP support")
 if(PROGRESS_OPENMP OR BML_OPENMP)