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

Add test_cpp_extensions tests for stream_and_event and mita_backend (#123614)

Browse Source 
Test the generic torch.Stream/Event with fake device gurad and hooks. Since we added a fake device backend, it is mutual exclusive to other backends. Tests will be skipped if TEST_CUDA or TEST_ROCM is true.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/123614
Approved by: https://github.com/albanD
ghstack dependencies: #123611, #123612
This commit is contained in:
egienvalue
2024-04-25 11:48:49 -07:00
committed by PyTorch MergeBot
parent 73744a2c00
commit 8461e7ed9e
5 changed files with 487 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tools/testing/modulefinder_determinator.py
View File

@ -21,6 +21,8 @@ TARGET_DET_LIST = [
"test_cpp_extensions_aot_no_ninja",
"test_cpp_extensions_jit",
"test_cpp_extensions_open_device_registration",
"test_cpp_extensions_stream_and_event",
"test_cpp_extensions_mtia_backend",
"test_cuda",
"test_cuda_primary_ctx",
"test_dataloader",