add default config benchmarks for ampere

hpcgarage · Nov 3, 2021 · 7964328 · 7964328
1 parent 4de2c56
commit 7964328
Show file tree

Hide file tree

Showing 2 changed files with 4,388 additions and 4 deletions.
diff --git a/bench/device/CMakeLists.txt b/bench/device/CMakeLists.txt
@@ -1,16 +1,34 @@
-# Just the default configurations
-add_executable(cuasr_bench_srgemm_device_defaults
+# SM50 defualt configurations
+add_executable(cuasr_bench_srgemm_device_sm50_defaults
   sm50_defaults.cu
 )
 target_include_directories(
-  cuasr_bench_srgemm_device_defaults
+  cuasr_bench_srgemm_device_sm50_defaults
   PRIVATE
   ${PROJECT_SOURCE_DIR}/include/
   ${PROJECT_SOURCE_DIR}/tools/include/
   ${PROJECT_SOURCE_DIR}/cutlass/include/
   ${PROJECT_SOURCE_DIR}/cutlass/tools/util/include/
 )
-target_link_libraries(cuasr_bench_srgemm_device_defaults
+target_link_libraries(cuasr_bench_srgemm_device_sm50_defaults
+  benchmark
+  benchmark_main
+  ${cuASR_LIB_NAME}
+)
+
+# SM80 defualt configurations
+add_executable(cuasr_bench_srgemm_device_sm80_defaults
+  sm80_defaults.cu
+)
+target_include_directories(
+  cuasr_bench_srgemm_device_sm80_defaults
+  PRIVATE
+  ${PROJECT_SOURCE_DIR}/include/
+  ${PROJECT_SOURCE_DIR}/tools/include/
+  ${PROJECT_SOURCE_DIR}/cutlass/include/
+  ${PROJECT_SOURCE_DIR}/cutlass/tools/util/include/
+)
+target_link_libraries(cuasr_bench_srgemm_device_sm80_defaults
   benchmark
   benchmark_main
   ${cuASR_LIB_NAME}