Home
last modified time | relevance | path

Searched refs:streams1 (Results 1 – 4 of 4) sorted by relevance

/external/webrtc/net/dcsctp/rx/
Dtraditional_reassembly_streams_test.cc157 TraditionalReassemblyStreams streams1("", on_assembled.AsStdFunction()); in TEST_F() local
158 EXPECT_TRUE(streams1.GetHandoverReadiness().IsReady()); in TEST_F()
161 streams1.AddHandoverState(state); in TEST_F()
176 TraditionalReassemblyStreams streams1("", on_assembled.AsStdFunction()); in TEST_F() local
178 EXPECT_EQ(streams1.Add(tsn(1), gen_.Ordered({1}, "B")), 1); in TEST_F()
179 EXPECT_EQ(streams1.GetHandoverReadiness(), in TEST_F()
182 EXPECT_EQ(streams1.Add(tsn(2), gen_.Ordered({2, 3, 4})), 3); in TEST_F()
183 EXPECT_EQ(streams1.GetHandoverReadiness(), in TEST_F()
186 EXPECT_EQ(streams1.Add(tsn(3), gen_.Ordered({5, 6})), 2); in TEST_F()
187 EXPECT_EQ(streams1.GetHandoverReadiness(), in TEST_F()
[all …]
/external/pytorch/c10/xpu/test/impl/
DXPUGuardTest.cpp35 std::vector<c10::xpu::XPUStream> streams1; in TEST() local
39 streams1.push_back(c10::xpu::getStreamFromPool()); in TEST()
40 streams1.push_back(c10::xpu::getStreamFromPool()); in TEST()
43 EXPECT_EQ(streams1[0].device_index(), 1); in TEST()
44 EXPECT_EQ(streams1[1].device_index(), 1); in TEST()
/external/pytorch/aten/src/ATen/test/
Dcuda_stream_test.cpp134 std::vector<at::cuda::CUDAStream> streams1; in TEST() local
137 streams1.push_back(at::cuda::getDefaultCUDAStream()); in TEST()
138 streams1.push_back(at::cuda::getStreamFromPool()); in TEST()
140 ASSERT_EQ_CUDA(streams1[0].device_index(), 1); in TEST()
141 ASSERT_EQ_CUDA(streams1[1].device_index(), 1); in TEST()
142 at::cuda::setCurrentCUDAStream(streams1[0]); in TEST()
150 at::cuda::CUDAStreamGuard guard(streams1[1]); in TEST()
153 ASSERT_EQ_CUDA(at::cuda::getCurrentCUDAStream(1), streams1[1]); in TEST()
158 ASSERT_EQ_CUDA(at::cuda::getCurrentCUDAStream(1), streams1[0]); in TEST()
165 ASSERT_EQ_CUDA(at::cuda::getCurrentCUDAStream(1), streams1[0]); in TEST()
/external/pytorch/docs/cpp/source/notes/
Dtensor_cuda_stream.rst233 // create a CUDA stream vector `streams1` on device using CUDA device guard
234 std::vector<at::cuda::CUDAStream> streams1;
238 streams1.push_back(at::cuda::getDefaultCUDAStream());
239 streams1.push_back(at::cuda::getStreamFromPool());
243 // set current stream as `streams1[0]` on device 1
244 at::cuda::setCurrentCUDAStream(streams1[0]);
252 // current CUDA stream is still `streams1[0]` on device 1, no change
259 at::cuda::CUDAStreamGuard stream_guard(streams1[1]);
261 // current device index and current CUDA stream are set to 1 and `streams1[1]` within scope
270 at::cuda::CUDAMultiStreamGuard multi_guard({streams0[1], streams1[1]});
[all …]