Added a break statement after the Navi arch was found in the first occurrence.

akashveramd · akashveramd · commit 72dc0f072a8a · 2025-07-19T04:44:38.000Z
diff --git a/aten/src/ATen/native/cpu/AmpGradScalerKernels.cpp b/aten/src/ATen/native/cpu/AmpGradScalerKernels.cpp
@@ -55,7 +55,10 @@ void _amp_foreach_non_finite_check_and_unscale_cpu_kernel(
   for (auto index: c10::irange(detail::getCUDAHooks().deviceCount()))
   {
     if (detail::getCUDAHooks().isGPUArch(index, archs))
+    {
       is_navi_arch = true;
+      break;
+    }
   }
 
   // Ensures client code (GradScaler) filtered scaled_grads by dtype.

Original file line number	Diff line number	Diff line change
`@@ -55,7 +55,10 @@ void _amp_foreach_non_finite_check_and_unscale_cpu_kernel(`
`55`	`55`	`for (auto index: c10::irange(detail::getCUDAHooks().deviceCount()))`
`56`	`56`	`{`
`57`	`57`	`if (detail::getCUDAHooks().isGPUArch(index, archs))`
	`58`	`+ {`
`58`	`59`	`is_navi_arch = true;`
	`60`	`+ break;`
	`61`	`+ }`
`59`	`62`	`}`
`60`	`63`
`61`	`64`	`// Ensures client code (GradScaler) filtered scaled_grads by dtype.`