prevent crash when trying to fill an empty tensor

ngc92 · ngc92 · commit 8b14573b4b96 · 2026-03-06T22:24:50.000+01:00
diff --git a/src/kernels/fill.cu b/src/kernels/fill.cu
@@ -17,6 +17,8 @@ __global__ void fill_kernel(floatX* dst, floatX value, std::size_t count) {
 
 template<typename floatX>
 void fill_imp(floatX* dst, floatX value, std::size_t count, cudaStream_t stream) {
+    if (count == 0) return;
+    if (dst == nullptr) throw std::invalid_argument("dst is nullptr");
     fill_kernel<<<div_ceil(count, static_cast<std::size_t>(256)), 256, 0, stream>>> (dst, value, count);
     CUDA_CHECK(cudaGetLastError());
 }

Original file line number	Diff line number	Diff line change
`@@ -17,6 +17,8 @@ __global__ void fill_kernel(floatX* dst, floatX value, std::size_t count) {`
`17`	`17`
`18`	`18`	`template<typename floatX>`
`19`	`19`	`void fill_imp(floatX* dst, floatX value, std::size_t count, cudaStream_t stream) {`
	`20`	`+ if (count == 0) return;`
	`21`	`+ if (dst == nullptr) throw std::invalid_argument("dst is nullptr");`
`20`	`22`	`fill_kernel<<<div_ceil(count, static_cast<std::size_t>(256)), 256, 0, stream>>> (dst, value, count);`
`21`	`23`	`CUDA_CHECK(cudaGetLastError());`
`22`	`24`	`}`