Remove internal use of gpu_id. (#9568)
This commit is contained in:
@@ -12,7 +12,7 @@ namespace xgboost::linalg {
|
||||
namespace {
|
||||
void TestElementWiseKernel() {
|
||||
auto device = DeviceOrd::CUDA(0);
|
||||
Tensor<float, 3> l{{2, 3, 4}, 0};
|
||||
Tensor<float, 3> l{{2, 3, 4}, device};
|
||||
{
|
||||
/**
|
||||
* Non-contiguous
|
||||
|
||||
Reference in New Issue
Block a user