add dnnl type

ROCm · Jan 24, 2025 · d54a4bc · d54a4bc
1 parent abd877f
commit d54a4bc
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/src/targets/cpu/dnnl.cpp b/src/targets/cpu/dnnl.cpp
@@ -69,6 +69,7 @@ dnnl::memory::data_type to_dnnl_memory_data_type(shape::type_t t)
     case st::int8_type: return dt::s8;
     case st::uint8_type: return dt::u8;
     case st::fp8e4m3fnuz_type: MIGRAPHX_THROW("fp8e4m3fnuz unsupported in DNNL");
+    case st::bf16_type: return dt::bf16;
     default: MIGRAPHX_THROW("Unsupported data type");
     }
 }