"git@developer.sourcefind.cn:modelzoo/resnet50_tensorflow.git" did not exist on "1307077a49d8620646b09a15afd67d0d16122ba0"
Commit bd689f40 authored by illsilin's avatar illsilin
Browse files

merge from public repo

parents c160c6cf a94113a9
This diff is collapsed.
This diff is collapsed.
...@@ -66,6 +66,12 @@ inline bool is_lds_direct_load_supported() ...@@ -66,6 +66,12 @@ inline bool is_lds_direct_load_supported()
ck::get_device_name() == "gfx950"; ck::get_device_name() == "gfx950";
} }
inline bool is_bf16_atomic_supported()
{
return ck::get_device_name() == "gfx940" || ck::get_device_name() == "gfx941" ||
ck::get_device_name() == "gfx942";
}
inline bool is_gfx101_supported() inline bool is_gfx101_supported()
{ {
return ck::get_device_name() == "gfx1010" || ck::get_device_name() == "gfx1011" || return ck::get_device_name() == "gfx1010" || ck::get_device_name() == "gfx1011" ||
......
This diff is collapsed.
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