Add Ampere GEMM example in Cute and CUTLASS 3.x

codeplaysoftware · Aug 2, 2024 · 4c0c974 · 4c0c974
1 parent 8c5841c
commit 4c0c974
Show file tree

Hide file tree

Showing 2 changed files with 428 additions and 256 deletions.
diff --git a/examples/14_ampere_tf32_tensorop_gemm/CMakeLists.txt b/examples/14_ampere_tf32_tensorop_gemm/CMakeLists.txt
@@ -27,22 +27,19 @@
 # OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
 # OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 
-
 if (CUTLASS_ENABLE_SYCL)
-  if (SYCL_NVIDIA_TARGET)
-    cutlass_example_add_executable(
-      14_ampere_tf32_tensorop_gemm_cute
-      ampere_tf32_tensorop_gemm_cute.cpp
+  cutlass_example_add_executable(
+    14_ampere_tf32_tensorop_gemm_cute
+    ampere_tf32_tensorop_gemm_cute.cu
     )
-  endif()
 else()
   cutlass_example_add_executable(
     14_ampere_tf32_tensorop_gemm
     ampere_tf32_tensorop_gemm.cu
-  )
+    )
 
   cutlass_example_add_executable(
     14_ampere_tf32_tensorop_gemm_cute
     ampere_tf32_tensorop_gemm_cute.cu
-  )
+    )
 endif()