[Batchnorm][ASM] Disabling ASM kernel for unsupported target (gfx940)…

… (#1312)
ROCm · Dec 4, 2021 · cd0d942 · cd0d942
1 parent 330bdd9
commit cd0d942
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/src/solver/batchnorm/backward_spatial_single.cpp b/src/solver/batchnorm/backward_spatial_single.cpp
@@ -210,9 +210,11 @@ BnBwdTrainingSpatialSingle::GetSolution(const ExecutionContext& context,
            (StartsWith(handle.GetDeviceName(), "gfx8") ||
             (StartsWith(handle.GetDeviceName(), "gfx9")
 #if WORKAROUND_ISSUE_1146
-             && (handle.GetDeviceName() != "gfx90a")
+             &&
+             (handle.GetDeviceName() != "gfx90a")
 #endif
-                 )) &&
+             &&
+             (handle.GetDeviceName() != "gfx940"))) &&
            (!handle.GetTargetProperties().Xnack() || !*handle.GetTargetProperties().Xnack()))
         {
             kernel.kernel_file = "gcnAsmBNBwdTrainSpatial.s";