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

[Inductor][CPP] Fix the test case of test_linear_reuse_kernels (#163723)

Browse Source 
Fixes #163491.
Add tolerances to make `test_linear_reuse_kernels` more stable.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/163723
Approved by: https://github.com/leslie-fang-intel
This commit is contained in:
CaoE
2025-09-29 05:29:01 +00:00
committed by PyTorch MergeBot
parent eb4361a801
commit b9854c9d89
1 changed files with 14 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
test/inductor/test_cpu_select_algorithm.py
View File

@ -2936,10 +2936,20 @@ class TestSelectAlgorithm(BaseTestSelectAlgorithm):
x = torch.randn(batch_size, in_features).to(dtype=dtype) x = torch.randn(batch_size, in_features).to(dtype=dtype)
mod = M().to(dtype=dtype).eval() mod = M().to(dtype=dtype).eval()
self.common(mod, (x)) with verify(dtype) as (atol, rtol):
_, code = run_and_get_cpp_code(mod, x) ref_res = mod(x)
# Check that only 2 kernels are in the generated code m = torch.compile(mod)
assert code.count("AMXState amx_state") == 2 res, code = run_and_get_cpp_code(m, x)
self.assertEqual(
res,
ref_res,
atol=atol,
rtol=rtol,
equal_nan=True,
exact_dtype=True,
)
# Check that only 2 kernels are in the generated code
assert code.count("AMXState amx_state") == 2
@dynamo_config.patch({"dynamic_shapes": True, "assume_static_by_default": False}) @dynamo_config.patch({"dynamic_shapes": True, "assume_static_by_default": False})