File tree 1 file changed +5
-2
lines changed
1 file changed +5
-2
lines changed Original file line number Diff line number Diff line change @@ -536,7 +536,7 @@ InvokerFactory MakeImplGemmDynamicForwardXdlopsNHWCInvokerFactory(
536
536
if (problem.GetOut ().GetType () == miopenHalf)
537
537
return use_fp32_global_split_on_fp16;
538
538
if (problem.GetOut ().GetType () == miopenBFloat16)
539
- return need_set_zero ;
539
+ return config. gemm_k_global_split > 0 ;
540
540
return false ;
541
541
}();
542
542
const auto is_nchw = problem.IsLayoutDefault ();
@@ -849,7 +849,10 @@ InvokerFactory MakeImplGemmDynamicBackwardDataXdlopsNHWCInvokerFactory(
849
849
if (problem.GetOut ().GetType () == miopenHalf)
850
850
return use_fp32_global_split_on_fp16;
851
851
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
+ }
853
856
return false ;
854
857
}();
855
858
const auto is_nchw = problem.IsLayoutDefault ();
You can’t perform that action at this time.
0 commit comments