We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 7494259 commit 560b376Copy full SHA for 560b376
src/torchcodec/_core/BetaCudaDeviceInterface.cpp
@@ -152,6 +152,9 @@ BetaCudaDeviceInterface::~BetaCudaDeviceInterface() {
152
}
153
154
void BetaCudaDeviceInterface::initializeInterface(AVStream* avStream) {
155
+ torch::Tensor dummyTensorForCudaInitialization = torch::empty(
156
+ {1}, torch::TensorOptions().dtype(torch::kUInt8).device(device_));
157
+
158
TORCH_CHECK(avStream != nullptr, "AVStream cannot be null");
159
timeBase_ = avStream->time_base;
160
0 commit comments