Revert "[Batchnorm][ASM] Disabling ASM kernel for unsupported target …

…(gfx940) (#1312)" This reverts commit cd0d942.
ROCm · Dec 6, 2021 · 81995f4 · 81995f4
1 parent 62a3b2d
commit 81995f4
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/src/solver/batchnorm/backward_spatial_single.cpp b/src/solver/batchnorm/backward_spatial_single.cpp
@@ -210,11 +210,9 @@ 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";