Skip to content

GitLab

Commit 298e80c7 authored by Po Yen Chen's avatar Po Yen Chen
Browse files

Enable gfx941 support for DeviceGemmXdl<> device op

parent 15368549
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
include/ck/tensor_operation/gpu/device/impl/device_gemm_xdl.hpp
View file @ 298e80c7
...@@ -185,7 +185,7 @@ struct DeviceGemmXdl : public DeviceGemm<ALayout, ...@@ -185,7 +185,7 @@ struct DeviceGemmXdl : public DeviceGemm<ALayout,
} }
} }
else if(ck::get_device_name() == "gfx90a" || ck::get_device_name() == "gfx940" || else if(ck::get_device_name() == "gfx90a" || ck::get_device_name() == "gfx940" ||
ck::get_device_name() == "gfx942") ck::get_device_name() == "gfx941" || ck::get_device_name() == "gfx942")
{ {
if constexpr(!(is_same_v<AccDataType, float> || is_same_v<AccDataType, float> || if constexpr(!(is_same_v<AccDataType, float> || is_same_v<AccDataType, float> ||
is_same_v<AccDataType, int32_t> || is_same_v<AccDataType, double>)) is_same_v<AccDataType, int32_t> || is_same_v<AccDataType, double>))
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment