Skip to content

Commit d962f33

Browse files
committed
use cudaStreamCreate
1 parent acea41c commit d962f33

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

onnxruntime/test/shared_lib/test_inference.cc

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2844,7 +2844,7 @@ TEST(CApiTest, TestExternalCUDAStreamWithIOBinding) {
28442844
// updating provider option with user provided compute stream
28452845
cudaStream_t compute_stream = nullptr;
28462846
void* user_compute_stream = nullptr;
2847-
cudaStreamCreateWithFlags(&compute_stream, cudaStreamNonBlocking);
2847+
cudaStreamCreate(&compute_stream);
28482848
ASSERT_TRUE(api.UpdateTensorRTProviderOptionsWithValue(rel_trt_options.get(), "user_compute_stream", compute_stream) == nullptr);
28492849
ASSERT_TRUE(api.GetTensorRTProviderOptionsByName(rel_trt_options.get(), "user_compute_stream", &user_compute_stream) == nullptr);
28502850
ASSERT_TRUE(user_compute_stream == (void*)compute_stream);

0 commit comments

Comments
 (0)