File tree Expand file tree Collapse file tree 3 files changed +3
-20
lines changed Expand file tree Collapse file tree 3 files changed +3
-20
lines changed Original file line number Diff line number Diff line change @@ -462,10 +462,4 @@ void miopen::ConvolutionContext::DetectRocm()
462
462
use_binaries = !miopen::IsDisabled (MIOPEN_DEBUG_AMD_ROCM_PRECOMPILED_BINARIES{});
463
463
#endif
464
464
}
465
-
466
- if (StartsWith (GetStream ().GetDeviceName (), " gfx8" ))
467
- {
468
- use_asm_kernels = false ;
469
- use_binaries = false ;
470
- }
471
465
}
Original file line number Diff line number Diff line change @@ -186,12 +186,9 @@ bool ConvAsm3x3U::IsApplicable(const ConvolutionContext& params) const
186
186
return false ;
187
187
if (!params.rmv .IsV2orV3 ())
188
188
return false ;
189
-
190
189
const std::string name = params.GetStream ().GetDeviceName ();
191
- if (name.find (" gfx9" ) == std::string::npos)
192
- {
190
+ if (!(StartsWith (name, " gfx8" ) || StartsWith (name, " gfx9" )))
193
191
return false ;
194
- }
195
192
assert (params.weights_layout .length () == 0 ); // FIXME _weights_layout is not supported yet.
196
193
// clang-format off
197
194
return params.pad_w == 1
Original file line number Diff line number Diff line change @@ -344,12 +344,9 @@ bool ConvAsmBwdWrW3x3::IsApplicable(const ConvolutionContext& params) const
344
344
return false ;
345
345
if (!params.rmv .IsV2orV3 ())
346
346
return false ;
347
-
348
347
const std::string name = params.GetStream ().GetDeviceName ();
349
- if (name.find (" gfx9" ) == std::string::npos)
350
- {
348
+ if (!(StartsWith (name, " gfx8" ) || StartsWith (name, " gfx9" )))
351
349
return false ;
352
- }
353
350
assert (params.weights_layout .length () == 0 ); // _weights_layout is not supported yet
354
351
// clang-format off
355
352
bool ok = params.pad_w == 1 // -q pad_w
@@ -364,16 +361,12 @@ bool ConvAsmBwdWrW3x3::IsApplicable(const ConvolutionContext& params) const
364
361
&& (params.IsFp32 () || params.IsFp16 ())
365
362
&& params.in_layout == " NCHW" ;
366
363
if (!ok)
367
- {
368
364
return false ; // Early exit to speed up the check.
369
- }
370
365
371
366
if (params.IsFp16 ()
372
- && (name. find ( " gfx8" ) != std::string::npos // Not supported.
367
+ && (StartsWith (name, " gfx8" ) // Not supported.
373
368
|| params.batch_sz % 2 != 0 )) // / \todo Initial version.
374
- {
375
369
return false ;
376
- }
377
370
378
371
// Check limits:
379
372
const auto h_w = static_cast <long >(params.out_height ) * params.out_width ;
@@ -401,7 +394,6 @@ bool ConvAsmBwdWrW3x3::IsApplicable(const ConvolutionContext& params) const
401
394
&& n_c_h_w < std::pow (2 , 29 )
402
395
&& n_k_h_w < std::pow (2 , 29 )
403
396
&& c_k_r_s < std::pow (2 , 29 ); // clang-format on
404
-
405
397
return ok;
406
398
}
407
399
You can’t perform that action at this time.
0 commit comments