Allow for sympy.Expr in tensor lowering in inductor (pytorch#103678)

Signed-off-by: Edward Z. Yang <[email protected]> Pull Request resolved: pytorch#103678 Approved by: https://github.com/voznesenskym
juhofuriosa · Jun 16, 2023 · 39ba2e6 · 39ba2e6
1 parent dad29f9
commit 39ba2e6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/torch/_inductor/lowering.py b/torch/_inductor/lowering.py
@@ -1828,7 +1828,7 @@ def tensor(data, *, dtype=None, device=None, layout=None, pin_memory=False):
     else:
         dtype = dtype or torch.get_default_dtype()
 
-    if isinstance(data, sympy.Symbol):
+    if isinstance(data, sympy.Expr):
         ranges = []
 
         def inner_fn(index):