From 00db8f6b43bba0fa2b033afd45268417a86c0de7 Mon Sep 17 00:00:00 2001 From: Alejandro Acosta Date: Mon, 20 May 2024 14:34:24 +0100 Subject: [PATCH] Run sycl ampere example on cuda --- examples/CMakeLists.txt | 1 + examples/sycl/CMakeLists.txt | 2 +- examples/sycl/ampere/CMakeLists.txt | 4 ++-- ..._fp32.cpp => ampere_gemm_bf16_bf16_fp32_tensor_op_fp32.cu} | 0 ..._fp32.cpp => ampere_gemm_fp16_fp16_fp32_tensor_op_fp32.cu} | 0 5 files changed, 4 insertions(+), 3 deletions(-) rename examples/sycl/ampere/{ampere_gemm_bf16_bf16_fp32_tensor_op_fp32.cpp => ampere_gemm_bf16_bf16_fp32_tensor_op_fp32.cu} (100%) rename examples/sycl/ampere/{ampere_gemm_fp16_fp16_fp32_tensor_op_fp32.cpp => ampere_gemm_fp16_fp16_fp32_tensor_op_fp32.cu} (100%) diff --git a/examples/CMakeLists.txt b/examples/CMakeLists.txt index dd465ac1f..67e9ca0f2 100644 --- a/examples/CMakeLists.txt +++ b/examples/CMakeLists.txt @@ -164,6 +164,7 @@ else() 57_hopper_grouped_gemm 58_ada_fp8_gemm 59_ampere_gather_scatter_conv + sycl ) add_subdirectory(${EXAMPLE}) diff --git a/examples/sycl/CMakeLists.txt b/examples/sycl/CMakeLists.txt index 8dac4f6c5..ddc98f12c 100644 --- a/examples/sycl/CMakeLists.txt +++ b/examples/sycl/CMakeLists.txt @@ -29,6 +29,6 @@ if(SYCL_INTEL_TARGET) add_subdirectory(pvc) -else(SYCL_NVIDIA_TARGET) +else() add_subdirectory(ampere) endif() diff --git a/examples/sycl/ampere/CMakeLists.txt b/examples/sycl/ampere/CMakeLists.txt index 81c8545ec..58501f5e8 100644 --- a/examples/sycl/ampere/CMakeLists.txt +++ b/examples/sycl/ampere/CMakeLists.txt @@ -29,10 +29,10 @@ cutlass_example_add_executable( ampere_gemm_fp16_fp16_fp32_tensor_op_fp32 - ampere_gemm_fp16_fp16_fp32_tensor_op_fp32.cpp + ampere_gemm_fp16_fp16_fp32_tensor_op_fp32.cu ) cutlass_example_add_executable( ampere_gemm_bf16_bf16_fp32_tensor_op_fp32 - ampere_gemm_bf16_bf16_fp32_tensor_op_fp32.cpp + ampere_gemm_bf16_bf16_fp32_tensor_op_fp32.cu ) diff --git a/examples/sycl/ampere/ampere_gemm_bf16_bf16_fp32_tensor_op_fp32.cpp b/examples/sycl/ampere/ampere_gemm_bf16_bf16_fp32_tensor_op_fp32.cu similarity index 100% rename from examples/sycl/ampere/ampere_gemm_bf16_bf16_fp32_tensor_op_fp32.cpp rename to examples/sycl/ampere/ampere_gemm_bf16_bf16_fp32_tensor_op_fp32.cu diff --git a/examples/sycl/ampere/ampere_gemm_fp16_fp16_fp32_tensor_op_fp32.cpp b/examples/sycl/ampere/ampere_gemm_fp16_fp16_fp32_tensor_op_fp32.cu similarity index 100% rename from examples/sycl/ampere/ampere_gemm_fp16_fp16_fp32_tensor_op_fp32.cpp rename to examples/sycl/ampere/ampere_gemm_fp16_fp16_fp32_tensor_op_fp32.cu