hardcode compute type

a8409d7f · Khalique Ahmed · 00d5d880 · a8409d7f
Commit a8409d7f authored Jul 12, 2021 by Khalique Ahmed
Hide whitespace changes
Inline Side-by-side

Showing with 6 additions and 4 deletions

src/targets/gpu/gemm_impl.cpp src/targets/gpu/gemm_impl.cpp +6 -4

No files found.
--- a/src/targets/gpu/gemm_impl.cpp
+++ b/src/targets/gpu/gemm_impl.cpp
@@ -65,11 +65,13 @@ void gemm_impl(context& ctx,
        output_type = rocblas_datatype_i32_r;
    }
    auto compute_type = output_type;
-    if(ctx.get_stream().get_device_name() == "gfx908")
-    {
-        if(args[0].get_shape().type() == shape::half_type)
+    if(args[0].get_shape().type() == shape::half_type)
            compute_type = rocblas_datatype_f32_r;
-    }
+    // if(ctx.get_stream().get_device_name() == "gfx908")
+    // {
+    //     if(args[0].get_shape().type() == shape::half_type)
+    //         compute_type = rocblas_datatype_f32_r;
+    // }

 #if ROCBLAS_VERSION_MAJOR >= 2 && ROCBLAS_VERSION_MINOR >= 38
    rocblas_gemm_flags flag =