[TensorExpr] Fix lowering for aten::t. (#65859)

Summary: Pull Request resolved: #65859 Test Plan: Imported from OSS Reviewed By: VitalyFedyunin Differential Revision: D31289347 Pulled By: ZolotukhinM fbshipit-source-id: b9648416238657fe23366928e43ed63e992a8973
pytorch · Oct 12, 2021 · 5f15186 · 5f15186
1 parent 6864146
commit 5f15186
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 4 deletions.
diff --git a/test/test_jit_fuser_te.py b/test/test_jit_fuser_te.py
@@ -2039,6 +2039,7 @@ def bn_neither(i, x):
     'sqrt',
     'sub',
     'sum',
+    't',
     'tan',
     'tanh',
     'transpose',
@@ -2058,9 +2059,6 @@ def bn_neither(i, x):
 
 # If your OpInfo test causes this test to fail, add it here
 skip_ops = [
-    # Causing SIGSEGV
-    # Reference: https://github.com/pytorch/pytorch/pull/59442/checks?check_run_id=2746156896
-    't',
     'conj'
 ]
 

diff --git a/torch/csrc/jit/tensorexpr/operators/misc.cpp b/torch/csrc/jit/tensorexpr/operators/misc.cpp
@@ -344,7 +344,7 @@ Tensor computeTranspose(
     at::Device device) {
   auto A = c10::get<BufHandle>(inputs[0]);
   // Trivial case of 0-dim and 1-dim tensors: transpose is just a copy
-  if (A.ndim() < 1) {
+  if (A.ndim() <= 1) {
     return Compute(
         "aten_transpose",
         c10::fmap<DimArg>(outputShape),