Skip to content

Commit 1ee673c

Browse files
Fix rmm macro call (rapidsai#18108)
This fixes a call to the RMM macro due to a recent change in that library that helped provide better debug info. Authors: - Paul Mattione (https://github.com/pmattione-nvidia) - GALI PREM SAGAR (https://github.com/galipremsagar) - Bradley Dice (https://github.com/bdice) Approvers: - Nghia Truong (https://github.com/ttnghia) - Bradley Dice (https://github.com/bdice) URL: rapidsai#18108
1 parent ea2bca3 commit 1ee673c

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

java/src/main/native/src/RmmJni.cpp

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,5 @@
11
/*
2-
* Copyright (c) 2019-2024, NVIDIA CORPORATION.
2+
* Copyright (c) 2019-2025, NVIDIA CORPORATION.
33
*
44
* Licensed under the Apache License, Version 2.0 (the "License");
55
* you may not use this file except in compliance with the License.
@@ -657,7 +657,7 @@ JNIEXPORT jlong JNICALL Java_ai_rapids_cudf_Rmm_allocCudaInternal(JNIEnv* env,
657657
try {
658658
cudf::jni::auto_set_device(env);
659659
void* ptr{nullptr};
660-
RMM_CUDA_TRY_ALLOC(cudaMalloc(&ptr, size));
660+
RMM_CUDA_TRY_ALLOC(cudaMalloc(&ptr, size), size);
661661
return reinterpret_cast<jlong>(ptr);
662662
}
663663
CATCH_STD(env, 0)

0 commit comments

Comments
 (0)