Revert "USE_TMA_DEFAULT=False"

whitneywhtsang · whitneywhtsang · commit bbf58de45d14 · 2025-11-20T19:45:39.000Z
This reverts commit 6d69871.
diff --git a/scripts/patch/166927.patch b/scripts/patch/166927.patch
@@ -44,7 +44,7 @@ index 1a72e279aab79..5a4ed7190ef46 100644
                  "num_buffers_warp_spec", num_buffers_warp_spec
              )
  
-+        USE_TMA_DEFAULT = False
++        USE_TMA_DEFAULT = bool(torch.xpu.is_available())
 +        # The shape dtype of tensor desc is i32
 +        if V.graph.sizevars.statically_known_true(
 +            seq_len_q > 2**31 - 1

Original file line number	Diff line number	Diff line change
`@@ -44,7 +44,7 @@ index 1a72e279aab79..5a4ed7190ef46 100644`
`44`	`44`	`"num_buffers_warp_spec", num_buffers_warp_spec`
`45`	`45`	`)`
`46`	`46`
`47`		`-+ USE_TMA_DEFAULT = False`
	`47`	`++ USE_TMA_DEFAULT = bool(torch.xpu.is_available())`
`48`	`48`	`+ # The shape dtype of tensor desc is i32`
`49`	`49`	`+ if V.graph.sizevars.statically_known_true(`
`50`	`50`	`+ seq_len_q > 2**31 - 1`