temp hack for multi GPUs
This commit is contained in:
parent
6bbca9a8b7
commit
32ae49ab92
@ -20,7 +20,10 @@ void ArrayInterfaceHandler::SyncCudaStream(std::int64_t stream) {
|
|||||||
* case where 0 might be given should either use None, 1, or 2 instead for
|
* case where 0 might be given should either use None, 1, or 2 instead for
|
||||||
* clarity.
|
* clarity.
|
||||||
*/
|
*/
|
||||||
|
/* ignored for HIP */
|
||||||
|
#if !defined(XGBOOST_USE_HIP)
|
||||||
LOG(FATAL) << "Invalid stream ID in array interface: " << stream;
|
LOG(FATAL) << "Invalid stream ID in array interface: " << stream;
|
||||||
|
#endif
|
||||||
case 1:
|
case 1:
|
||||||
// default legacy stream
|
// default legacy stream
|
||||||
break;
|
break;
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user