Check __cuda_array_interface__ instead of cupy class. (#9971)

* Now XGBoost can directly consume CUDA data from torch.
This commit is contained in:
Jiaming Yuan
2024-01-09 19:59:01 +08:00
committed by GitHub
parent 2f57bbde3c
commit 01c4711556
6 changed files with 31 additions and 29 deletions

View File

@@ -162,6 +162,8 @@ Support Matrix
+-------------------------+-----------+-------------------+-----------+-----------+--------------------+-------------+
| cupy.ndarray | T | T | T | T | T | T |
+-------------------------+-----------+-------------------+-----------+-----------+--------------------+-------------+
| torch.Tensor | T | T | T | T | T | T |
+-------------------------+-----------+-------------------+-----------+-----------+--------------------+-------------+
| dlpack | CPA | CPA | | CPA | FF | FF |
+-------------------------+-----------+-------------------+-----------+-----------+--------------------+-------------+
| datatable.Frame | T | FF | | NPA | FF | |