enable ROCm for jvm and R

2023-10-30 12:52:44 -07:00
parent 32ae49ab92
commit 40dc263602
9 changed files with 31 additions and 5 deletions
--- a/python-package/packager/build_config.py
+++ b/python-package/packager/build_config.py
@@ -15,6 +15,10 @@ class BuildConfiguration:  # pylint: disable=R0902
    use_cuda: bool = False
    # Whether to enable NCCL
    use_nccl: bool = False
+    # Whether to enable HIP 
+    use_hip: bool = False
+    # Whether to enable RCCL
+    use_rccl: bool = False
    # Whether to enable HDFS
    use_hdfs: bool = False
    # Whether to enable Azure Storage