Skip to content

Commit 80f52ce

Browse files
committed
Update on "[inductor] add lowering for repeat_interleave.Tensor with output size specified (#147160)"
[ghstack-poisoned]
2 parents 4310217 + 749cf84 commit 80f52ce

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

test/inductor/test_torchinductor.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -13644,7 +13644,8 @@ def f(input, repeats):
1364413644
# we don't lower when the cpp_wrapper is used because it cannot generate
1364513645
# proper examples during autotune
1364613646
can_lower = not config.cpp_wrapper
13647-
self.assertEqual("repeat_interleave.Tensor" in code, can_lower)
13647+
has_lowered "repeat_interleave.Tensor" not in code
13648+
self.assertEqual(has_lowered, can_lower)
1364813649

1364913650

1365013651
@dataclasses.dataclass

0 commit comments

Comments
 (0)