saymrwulf/pytorch
1
0
Fork 0
mirror of https://github.com/saymrwulf/pytorch.git synced 2026-05-14 20:57:59 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

[Fix][Profiler UT] Skip CPU for the UT test/profiler/test_execution_trace.py::test_execution_trace_with_pt2 (#142027)

Browse source 
[Fix] Skip CPU device for the UT `test_execution_trace_with_pt2`
skip CPU because triton is only for GPUs. This UT is designed to test profiling the triton kernels.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/142027
Approved by: https://github.com/aaronenyeshi
This commit is contained in:
Chen, Zejun 2024-12-10 09:29:16 +00:00 committed by PyTorch MergeBot
parent 3d04de6b2f
commit 5ba61d7fb8
1 changed files with 5 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
test/profiler/test_execution_trace.py
View file

@ -34,7 +34,10 @@ from torch.profiler import (
supported_activities,
)
from torch.testing._internal.common_cuda import TEST_CUDA
from torch.testing._internal.common_device_type import instantiate_device_type_tests
from torch.testing._internal.common_device_type import (
instantiate_device_type_tests,
skipCPUIf,
)
from torch.testing._internal.common_utils import (
IS_WINDOWS,
run_tests,
@ -248,6 +251,7 @@ class TestExecutionTrace(TestCase):
(not has_triton()) or (not TEST_CUDA and not TEST_XPU),
"need triton and device(CUDA or XPU) availability to run",
)
@skipCPUIf(True, "skip CPU device for testing profiling triton")
def test_execution_trace_with_pt2(self, device):
@torchdynamo.optimize("inductor")
def fn(a, b, c):