fbcode/libspdl/cuda/storage.cpp (facebookresearch#1603)

Summary: Pull Request resolved: facebookresearch#1603

Reviewed By: frbng

Differential Revision: D114253091

Author: DevmateHteNullableReturn Bot

src/libspdl/cuda/storage.cpp

@@ -51,6 +51,9 @@ uintptr_t default_allocator(int size, int device, uintptr_t) {
     TRACE_EVENT("decoding", "cudaMalloc");
     CHECK_CUDA(cudaMalloc(&data, size), "Failed to allocate CUDA memory");
   }
+  if (data == nullptr) {
+    SPDL_FAIL("CUDA memory allocation returned a null pointer.");
+  }
   return reinterpret_cast<uintptr_t>(data);
 }