Avoid divide by 0 when calculating predicted performance with streamk (#1373)

daineAMD · web-flow · commit d001ff3c66dc · 2024-11-26T13:42:21.000-07:00
diff --git a/tensilelite/Tensile/Source/lib/source/ContractionSolution.cpp b/tensilelite/Tensile/Source/lib/source/ContractionSolution.cpp
@@ -56,7 +56,7 @@ namespace TensileLite
             __device__ __host__ inline constexpr N safe_ceil_div(N n, D d)
             {
                 // Static cast to undo integral promotion.
-                return static_cast<N>(n / d + (n % d != 0 ? 1 : 0));
+                return static_cast<N>(d == 0 ? 0 : (n / d + (n % d != 0 ? 1 : 0)));
             }
         } // namespace math
 

Original file line number	Diff line number	Diff line change
`@@ -56,7 +56,7 @@ namespace TensileLite`
`56`	`56`	`__device__ __host__ inline constexpr N safe_ceil_div(N n, D d)`
`57`	`57`	`{`
`58`	`58`	`// Static cast to undo integral promotion.`
`59`		`- return static_cast<N>(n / d + (n % d != 0 ? 1 : 0));`
	`59`	`+ return static_cast<N>(d == 0 ? 0 : (n / d + (n % d != 0 ? 1 : 0)));`
`60`	`60`	`}`
`61`	`61`	`} // namespace math`
`62`	`62`