pytorch

c8579b73 - Run `test_cpp_memory_snapshot_pickle` only when linux and x86_64 (#101366)

Run `test_cpp_memory_snapshot_pickle` only when linux and x86_64 (#101366) On Arm, I got ``` Traceback (most recent call last): File "/opt/pytorch/pytorch/test/test_cuda.py", line 5260, in test_cpp_memory_snapshot_pickle mem = run() File "/opt/pytorch/pytorch/test/test_cuda.py", line 5257, in run t = the_script_fn() File "/usr/local/lib/python3.10/dist-packages/torch/testing/_internal/common_utils.py", line 496, in prof_func_call return prof_callable(func_call, *args, **kwargs) File "/usr/local/lib/python3.10/dist-packages/torch/testing/_internal/common_utils.py", line 493, in prof_callable return callable(*args, **kwargs) RuntimeError: The following operation failed in the TorchScript interpreter. Traceback of TorchScript (most recent call last): File "/opt/pytorch/pytorch/test/test_cuda.py", line 5254, in the_script_fn @torch.jit.script def the_script_fn(): return torch.rand(311, 411, device='cuda') ~~~~~~~~~~ <--- HERE RuntimeError: record_context_cpp is not support on non-linux non-x86_64 platforms ``` https://github.com/pytorch/pytorch/blob/dfe484a3b32aeaa41819f7d1a341ca34920b2559/torch/csrc/profiler/unwind/unwind.cpp#L4-L24 seems related Pull Request resolved: https://github.com/pytorch/pytorch/pull/101366 Approved by: https://github.com/zdevito