Skip to content

GitLab

Commit 57d0ea67 authored by Jing Zhang's avatar Jing Zhang
Browse files

add gfx908 and gfx90a

parent 747039cc
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
include/ck/tensor_operation/gpu/device/impl/device_gemm_dl.hpp
View file @ 57d0ea67
...@@ -285,6 +285,7 @@ struct DeviceGemmDl : public DeviceGemm<ALayout, ...@@ -285,6 +285,7 @@ struct DeviceGemmDl : public DeviceGemm<ALayout,
static bool IsSupportedArgument(const Argument& arg) static bool IsSupportedArgument(const Argument& arg)
{ {
if(ck::get_device_name() == "gfx906" || ck::get_device_name() == "gfx1030" || if(ck::get_device_name() == "gfx906" || ck::get_device_name() == "gfx1030" ||
ck::get_device_name() == "gfx908" || ck::get_device_name() == "gfx90a" ||
ck::get_device_name() == "gfx1100" || ck::get_device_name() == "gfx1101" || ck::get_device_name() == "gfx1100" || ck::get_device_name() == "gfx1101" ||
ck::get_device_name() == "gfx1102") ck::get_device_name() == "gfx1102")
{ {
......
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