[TensorExpr] Fuser: disallow aten::slice nodes. (#43365)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/43365
We don't have shape inference for them yet.
Test Plan: Imported from OSS
Reviewed By: eellison
Differential Revision: D23253418
Pulled By: ZolotukhinM
fbshipit-source-id: 9c38778b8a616e70f6b2cb5aab03d3c2013b34b0
diff --git a/test/test_tensorexpr.py b/test/test_tensorexpr.py
index a433f93..29a0c84 100644
--- a/test/test_tensorexpr.py
+++ b/test/test_tensorexpr.py
@@ -1072,6 +1072,7 @@
# r = test(x, y, z)
# assert llvm.elapsed_value == 1 or interp.elapsed_value() > 1
+ @unittest.skip("no shape inference for aten::slice yet")
def test_slice(self):
def easy(x, y):
a = x[0:512:2]
diff --git a/torch/csrc/jit/passes/tensorexpr_fuser.cpp b/torch/csrc/jit/passes/tensorexpr_fuser.cpp
index 5fbaecd..8e96b78 100644
--- a/torch/csrc/jit/passes/tensorexpr_fuser.cpp
+++ b/torch/csrc/jit/passes/tensorexpr_fuser.cpp
@@ -69,7 +69,6 @@
case aten::reciprocal:
case aten::expm1:
case aten::lgamma:
- case aten::slice:
case aten::unsqueeze:
case aten::frac:
// TODO: uncomment once we can handle rand+broadcasts
@@ -94,6 +93,9 @@
return false;
}
return true;
+ case aten::slice:
+ // TODO: Shape inference is not implemented for this op yet
+ return false;
default:
return false;
}
