frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 21:14:14 +08:00
Code Issues Packages Projects Releases Wiki Activity

enable test (#155342)

Browse Source 
Pull Request resolved: https://github.com/pytorch/pytorch/pull/155342
Approved by: https://github.com/Skylion007, https://github.com/bdhirsh
ghstack dependencies: #154768
This commit is contained in:
eellison
2025-06-06 09:59:53 -07:00
committed by PyTorch MergeBot
parent d4d0ede6ba
commit ea37f72099
2 changed files with 0 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
test/inductor/test_torchinductor.py
View File

@ -8276,7 +8276,6 @@ def forward(self, arg0_1: "Sym(s77)", arg1_1: "Sym(s27)", arg2_1: "Sym(s53)", ar
actual_out = compiled_fn(view)
self.assertEqual(reference_out.stride(), actual_out.stride())
@xfail_if_triton_cpu
def test_like_channels_last(self):
def foo():
randn = torch.randn((4, 3, 8, 8), device=self.device, dtype=torch.float32)

2
test/inductor/test_triton_cpu_backend.py
View File

@ -13,12 +13,10 @@ except ImportError:
if HAS_CPU and TRITON_HAS_CPU:
@config.patch(cpu_backend="triton")
@config.patch("triton.coalesce_tiling_analysis", False)
class SweepInputsCpuTritonTest(test_torchinductor.SweepInputsCpuTest):
pass
@config.patch(cpu_backend="triton")
@config.patch("triton.coalesce_tiling_analysis", False)
class CpuTritonTests(test_torchinductor.TestCase):
common = test_torchinductor.check_model
device = "cpu"