Hendrik/xgboost
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

enable ROCm for jvm and R

Browse Source
This commit is contained in:
Hui Liu
2023-10-30 12:52:44 -07:00
parent 32ae49ab92
commit 40dc263602
9 changed files with 31 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
jvm-packages/xgboost4j-gpu/pom.xml
View File

@@ -104,6 +104,8 @@
<argument>${log.capi.invocation}</argument>
<argument>--use-cuda</argument>
<argument>${use.cuda}</argument>
<argument>--use-hip</argument>
<argument>${use.hip}</argument>
</arguments>
<workingDirectory>${user.dir}</workingDirectory>
</configuration>

2
jvm-packages/xgboost4j-gpu/src/native/xgboost4j-gpu.cpp
View File

@@ -2,7 +2,7 @@
// Created by bobwang on 2021/9/8.
//
#ifndef XGBOOST_USE_CUDA
#if !defined(XGBOOST_USE_CUDA) && !defined(XGBOOST_USE_HIP)
#include <jni.h>

4
jvm-packages/xgboost4j-gpu/src/native/xgboost4j-gpu.cu
View File

@@ -1,6 +1,10 @@
#include <jni.h>
#if defined(XGBOOST_USE_CUDA)
#include "../../../../src/common/device_helpers.cuh"
#elif defined(XGBOOST_USE_HIP)
#include "../../../../src/common/device_helpers.hip.h"
#endif
#include "../../../../src/common/cuda_pinned_allocator.h"
#include "../../../../src/data/array_interface.h"
#include "jvm_utils.h"

4
jvm-packages/xgboost4j-gpu/src/native/xgboost4j-gpu.hip
View File

@@ -0,0 +1,4 @@
#if defined(XGBOOST_USE_HIP)
#include "xgboost4j-gpu.cu"
#endif