Skip to content

Commit 1959c24

Browse files
Update ggml/src/ggml-cuda/ggml-cuda.cu
Add a defensive security assert Co-authored-by: Johannes Gäßler <[email protected]>
1 parent bd21613 commit 1959c24

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

ggml/src/ggml-cuda/ggml-cuda.cu

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -247,6 +247,7 @@ static ggml_cuda_device_info ggml_cuda_init() {
247247
info.devices[id].nsm = prop.multiProcessorCount;
248248
info.devices[id].smpb = prop.sharedMemPerBlock;
249249
info.devices[id].warp_size = prop.warpSize;
250+
GGML_ASSERT(!integrated || prop.canUseHostPointerForRegisteredMem);
250251
#if defined(GGML_USE_HIP) && defined(__HIP_PLATFORM_AMD__)
251252
info.devices[id].smpbo = prop.sharedMemPerBlock;
252253

0 commit comments

Comments
 (0)