Merge branch 'develop' into mobilenet_fp8

a2f8e035 · Umang Yadav · a6731a3f · 6a72e8fc · a2f8e035
Commit a2f8e035 authored Dec 07, 2023 by Umang Yadav
Show whitespace changes
Inline Side-by-side

Showing with 7 additions and 0 deletions

src/targets/gpu/target.cpp src/targets/gpu/target.cpp +7 -0

No files found.
--- a/src/targets/gpu/target.cpp
+++ b/src/targets/gpu/target.cpp
@@ -119,6 +119,13 @@ std::vector<pass> target::get_passes(migraphx::context& gctx, const compile_opti
        unsupported_fp8_ops.insert("convolution");
        unsupported_fp8_ops.insert("quant_convolution");
    }
+    // MIOpen doesn't have support for fp8 pooling yet.
+    unsupported_fp8_ops.insert("pooling");
+    if(not gpu::gfx_has_fp8_intrinsics())
+    {
+        unsupported_fp8_ops.insert("convolution");
+        unsupported_fp8_ops.insert("quant_convolution");
+    }
    // add all device kernels
    unsupported_fp8_ops.insert("logsoftmax");
    unsupported_fp8_ops.insert("nonzero");