uncomment the stuff (getting ready for merge)

NVIDIA · Sep 19, 2023 · ba3f415 · ba3f415
1 parent 8b112f0
commit ba3f415
Show file tree

Hide file tree

Showing 3 changed files with 55 additions and 55 deletions.
diff --git a/test/unit/gemm/warp/CMakeLists.txt b/test/unit/gemm/warp/CMakeLists.txt
@@ -28,19 +28,19 @@
 
 cutlass_test_unit_add_executable(
   cutlass_test_unit_gemm_warp
-  #gemm_sm50.cu
-  #gemm_sm60.cu
-  #gemm_sm61.cu
-  #gemm_sm70.cu
-  #gemm_sm75.cu
-  #gemm_sm80.cu
-  #gemm_complex_sm80.cu
-  #gemm_sparse_sm80.cu
-  #gemm_gaussian_complex_sm80.cu
+  gemm_sm50.cu
+  gemm_sm60.cu
+  gemm_sm61.cu
+  gemm_sm70.cu
+  gemm_sm75.cu
+  gemm_sm80.cu
+  gemm_complex_sm80.cu
+  gemm_sparse_sm80.cu
+  gemm_gaussian_complex_sm80.cu
   gemm_mixed_input_sm80.cu
-  #gemm_sm90.cu
-  #gemm_complex_sm90.cu
-  #wmma_sm70.cu
-  #wmma_sm72.cu
-  #wmma_sm75.cu
+  gemm_sm90.cu
+  gemm_complex_sm90.cu
+  wmma_sm70.cu
+  wmma_sm72.cu
+  wmma_sm75.cu
 )
diff --git a/tools/library/CMakeLists.txt b/tools/library/CMakeLists.txt
@@ -67,19 +67,19 @@ cutlass_add_library(
   src/util.cu
 
   # files split for parallel compilation
-  #src/reference/gemm_int4.cu
-  #src/reference/gemm_int8_canonical.cu
-  #src/reference/gemm_int8_interleaved_32.cu
-  #src/reference/gemm_int8_interleaved_64.cu
-  #src/reference/gemm_e4m3a_e4m3out.cu
-  #src/reference/gemm_e5m2a_e4m3out.cu
-  #src/reference/gemm_e4m3a_e5m2out.cu
-  #src/reference/gemm_e5m2a_e5m2out.cu
-  #src/reference/gemm_fp8in_fp16out.cu
-  #src/reference/gemm_fp8in_bf16out.cu
-  #src/reference/gemm_fp8in_fp32out.cu
-  #src/reference/gemm_fp32out.cu
-  #src/reference/gemm_fp_other.cu
+  src/reference/gemm_int4.cu
+  src/reference/gemm_int8_canonical.cu
+  src/reference/gemm_int8_interleaved_32.cu
+  src/reference/gemm_int8_interleaved_64.cu
+  src/reference/gemm_e4m3a_e4m3out.cu
+  src/reference/gemm_e5m2a_e4m3out.cu
+  src/reference/gemm_e4m3a_e5m2out.cu
+  src/reference/gemm_e5m2a_e5m2out.cu
+  src/reference/gemm_fp8in_fp16out.cu
+  src/reference/gemm_fp8in_bf16out.cu
+  src/reference/gemm_fp8in_fp32out.cu
+  src/reference/gemm_fp32out.cu
+  src/reference/gemm_fp_other.cu
   src/reference/gemm_fp_mixed_input.cu
   src/reference/initialize_reference_operations.cu
 

diff --git a/tools/library/src/reference/initialize_reference_operations.cu b/tools/library/src/reference/initialize_reference_operations.cu
@@ -43,19 +43,19 @@ namespace cutlass {
 namespace library {
 
 // note: init methods for the same op-class may be split into multiple to parallelize compilation
-// void initialize_gemm_reference_operations_int4(Manifest &manifest);
-// void initialize_gemm_reference_operations_int8_interleaved_32(Manifest &manifest);
-// void initialize_gemm_reference_operations_int8_interleaved_64(Manifest &manifest);
-// void initialize_gemm_reference_operations_int8_canonical(Manifest &manifest);
-// void initialize_gemm_reference_operations_e4m3a_e4m3out(Manifest &manifest);
-// void initialize_gemm_reference_operations_e5m2a_e4m3out(Manifest &manifest);
-// void initialize_gemm_reference_operations_e4m3a_e5m2out(Manifest &manifest);
-// void initialize_gemm_reference_operations_e5m2a_e5m2out(Manifest &manifest);
-// void initialize_gemm_reference_operations_fp8in_fp16out(Manifest &manifest);
-// void initialize_gemm_reference_operations_fp8in_bf16out(Manifest &manifest);
-// void initialize_gemm_reference_operations_fp8in_fp32out(Manifest &manifest);
-// void initialize_gemm_reference_operations_fp32out(Manifest &manifest);
-// void initialize_gemm_reference_operations_fp_other(Manifest &manifest);
+void initialize_gemm_reference_operations_int4(Manifest &manifest);
+void initialize_gemm_reference_operations_int8_interleaved_32(Manifest &manifest);
+void initialize_gemm_reference_operations_int8_interleaved_64(Manifest &manifest);
+void initialize_gemm_reference_operations_int8_canonical(Manifest &manifest);
+void initialize_gemm_reference_operations_e4m3a_e4m3out(Manifest &manifest);
+void initialize_gemm_reference_operations_e5m2a_e4m3out(Manifest &manifest);
+void initialize_gemm_reference_operations_e4m3a_e5m2out(Manifest &manifest);
+void initialize_gemm_reference_operations_e5m2a_e5m2out(Manifest &manifest);
+void initialize_gemm_reference_operations_fp8in_fp16out(Manifest &manifest);
+void initialize_gemm_reference_operations_fp8in_bf16out(Manifest &manifest);
+void initialize_gemm_reference_operations_fp8in_fp32out(Manifest &manifest);
+void initialize_gemm_reference_operations_fp32out(Manifest &manifest);
+void initialize_gemm_reference_operations_fp_other(Manifest &manifest);
 void initialize_gemm_reference_operations_fp_mixed_input(Manifest &manifest);
 
 void initialize_conv2d_reference_operations(Manifest &manifest);
@@ -64,25 +64,25 @@ void initialize_conv3d_reference_operations(Manifest &manifest);
 ///////////////////////////////////////////////////////////////////////////////////////////////////
 
 void initialize_reference_operations(Manifest &manifest) {
-  // initialize_conv2d_reference_operations(manifest);
-  // initialize_conv3d_reference_operations(manifest);
+  initialize_conv2d_reference_operations(manifest);
+  initialize_conv3d_reference_operations(manifest);
 
-  // initialize_gemm_reference_operations_int4(manifest);
+  initialize_gemm_reference_operations_int4(manifest);
 
-  // initialize_gemm_reference_operations_int8_interleaved_32(manifest);
-  // initialize_gemm_reference_operations_int8_interleaved_64(manifest);
-  // initialize_gemm_reference_operations_int8_canonical(manifest);
+  initialize_gemm_reference_operations_int8_interleaved_32(manifest);
+  initialize_gemm_reference_operations_int8_interleaved_64(manifest);
+  initialize_gemm_reference_operations_int8_canonical(manifest);
 
-  // initialize_gemm_reference_operations_e4m3a_e4m3out(manifest);
-  // initialize_gemm_reference_operations_e5m2a_e4m3out(manifest);
-  // initialize_gemm_reference_operations_e4m3a_e5m2out(manifest);
-  // initialize_gemm_reference_operations_e5m2a_e5m2out(manifest);
-  // initialize_gemm_reference_operations_fp8in_fp16out(manifest);
-  // initialize_gemm_reference_operations_fp8in_bf16out(manifest);
-  // initialize_gemm_reference_operations_fp8in_fp32out(manifest);
+  initialize_gemm_reference_operations_e4m3a_e4m3out(manifest);
+  initialize_gemm_reference_operations_e5m2a_e4m3out(manifest);
+  initialize_gemm_reference_operations_e4m3a_e5m2out(manifest);
+  initialize_gemm_reference_operations_e5m2a_e5m2out(manifest);
+  initialize_gemm_reference_operations_fp8in_fp16out(manifest);
+  initialize_gemm_reference_operations_fp8in_bf16out(manifest);
+  initialize_gemm_reference_operations_fp8in_fp32out(manifest);
 
-  // initialize_gemm_reference_operations_fp32out(manifest);
-  // initialize_gemm_reference_operations_fp_other(manifest);
+  initialize_gemm_reference_operations_fp32out(manifest);
+  initialize_gemm_reference_operations_fp_other(manifest);
   initialize_gemm_reference_operations_fp_mixed_input(manifest);
 }