Skip to content

Commit d22d5a1

Browse files
authored
Merge pull request #3455 from ROCm/release-fixes/fix-misa-tensor-casting
Fix tensor casting happening in all cases (#3443)
2 parents 0a7edc6 + c512f3f commit d22d5a1

File tree

1 file changed

+5
-2
lines changed

1 file changed

+5
-2
lines changed

src/conv/invokers/impl_gemm_dynamic.cpp

+5-2
Original file line numberDiff line numberDiff line change
@@ -536,7 +536,7 @@ InvokerFactory MakeImplGemmDynamicForwardXdlopsNHWCInvokerFactory(
536536
if(problem.GetOut().GetType() == miopenHalf)
537537
return use_fp32_global_split_on_fp16;
538538
if(problem.GetOut().GetType() == miopenBFloat16)
539-
return need_set_zero;
539+
return config.gemm_k_global_split > 0;
540540
return false;
541541
}();
542542
const auto is_nchw = problem.IsLayoutDefault();
@@ -849,7 +849,10 @@ InvokerFactory MakeImplGemmDynamicBackwardDataXdlopsNHWCInvokerFactory(
849849
if(problem.GetOut().GetType() == miopenHalf)
850850
return use_fp32_global_split_on_fp16;
851851
if(problem.GetOut().GetType() == miopenBFloat16)
852-
return need_set_zero;
852+
{
853+
return (y < stride_h || x < stride_w || dilation_h != 1 || dilation_w != 1 ||
854+
config.gemm_k_global_split > 0);
855+
}
853856
return false;
854857
}();
855858
const auto is_nchw = problem.IsLayoutDefault();

0 commit comments

Comments
 (0)