diff --git a/svg/kmeans_utils.py b/svg/kmeans_utils.py index b2d5789..e2e3744 100755 --- a/svg/kmeans_utils.py +++ b/svg/kmeans_utils.py @@ -453,7 +453,7 @@ def _cfg_keep(conf): BN = conf.kwargs["BLOCK_N"] BK = conf.kwargs["BLOCK_K"] # Avoid tiny tiles on many warps - if BN * BK < 32 * 32 and conf.num_warps > 4: + if BN * BK <= 32 * 32 and conf.num_warps > 4: return False return True