add checkout dtype

xgqdut2016 · xgqdut2016 · commit b293febb46c4 · 2024-12-10T10:03:09.000+08:00
diff --git a/src/ops/random_sample/cuda/random_sample.cu b/src/ops/random_sample/cuda/random_sample.cu
@@ -141,6 +141,8 @@ void random_sample_workspace(size_t &size_radix_sort, size_t &size_scan,
             nullptr, size_scan,
             nullptr, voc,
             nullptr);
+    } else {
+        throw std::invalid_argument("Unsupported dtype provided.");
     }
 }
 __global__ void random_sample_kernel(uint64_t *result,

Original file line number	Diff line number	Diff line change
`@@ -141,6 +141,8 @@ void random_sample_workspace(size_t &size_radix_sort, size_t &size_scan,`
`141`	`141`	`nullptr, size_scan,`
`142`	`142`	`nullptr, voc,`
`143`	`143`	`nullptr);`
	`144`	`+ } else {`
	`145`	`+ throw std::invalid_argument("Unsupported dtype provided.");`
`144`	`146`	`}`
`145`	`147`	`}`
`146`	`148`	`__global__ void random_sample_kernel(uint64_t *result,`