From a5a35e71188ab1a20a53e4b4ea1e557aa2afdf03 Mon Sep 17 00:00:00 2001
From: DevmateHteNullableReturn Bot <noreply+1033302016106467@fb.com>
Date: Mon, 3 Aug 2026 12:38:06 -0700
Subject: [PATCH] fbcode/libspdl/cuda/storage.cpp (#1603)

Summary: Pull Request resolved: https://github.com/facebookresearch/spdl/pull/1603

Reviewed By: frbng

Differential Revision: D114253091
---
 src/libspdl/cuda/storage.cpp | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/src/libspdl/cuda/storage.cpp b/src/libspdl/cuda/storage.cpp
index af59e2546..82df51400 100644
--- a/src/libspdl/cuda/storage.cpp
+++ b/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);
 }