Home
last modified time | relevance | path

Searched refs:stream_set (Results 1 – 3 of 3) sorted by relevance

/external/pytorch/c10/xpu/
DXPUCachingAllocator.cpp30 using stream_set = ska::flat_hash_set<xpu::XPUStream>; typedef
45 stream_set stream_uses; // streams on which the block was used
392 stream_set streams(std::move(block->stream_uses)); in insert_events()
/external/pytorch/aten/src/ATen/test/
Dcuda_stream_test.cpp181 std::unordered_set<cudaStream_t> stream_set{}; in TEST() local
185 auto result_pair = stream_set.insert(cuda_stream); in TEST()
/external/pytorch/c10/cuda/
DCUDACachingAllocator.cpp136 using stream_set = ska::flat_hash_set<cuda::CUDAStream>; typedef
180 stream_set stream_uses; // streams on which the block was used
682 stream_set stream_uses = {};
1085 std::unordered_map<Block*, stream_set> block_to_cudagraph_stream_uses;
2993 stream_set streams(std::move(block->stream_uses)); in remove_cudagraph_stream_uses()
3009 stream_set streams(std::move(block->stream_uses)); in insert_events()