Skip to content

Commit

Permalink
Run sycl ampere example on cuda
Browse files Browse the repository at this point in the history
  • Loading branch information
aacostadiaz committed May 20, 2024
1 parent 7180fae commit 00db8f6
Show file tree
Hide file tree
Showing 5 changed files with 4 additions and 3 deletions.
1 change: 1 addition & 0 deletions examples/CMakeLists.txt
Original file line number Diff line number Diff line change
Expand Up @@ -164,6 +164,7 @@ else()
57_hopper_grouped_gemm
58_ada_fp8_gemm
59_ampere_gather_scatter_conv
sycl
)

add_subdirectory(${EXAMPLE})
Expand Down
2 changes: 1 addition & 1 deletion examples/sycl/CMakeLists.txt
Original file line number Diff line number Diff line change
Expand Up @@ -29,6 +29,6 @@

if(SYCL_INTEL_TARGET)
add_subdirectory(pvc)
else(SYCL_NVIDIA_TARGET)
else()
add_subdirectory(ampere)
endif()
4 changes: 2 additions & 2 deletions examples/sycl/ampere/CMakeLists.txt
Original file line number Diff line number Diff line change
Expand Up @@ -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
)

0 comments on commit 00db8f6

Please sign in to comment.