File tree Expand file tree Collapse file tree 2 files changed +2
-2
lines changed Expand file tree Collapse file tree 2 files changed +2
-2
lines changed Original file line number Diff line number Diff line change @@ -57,7 +57,7 @@ static std::unique_ptr<IDataTransfer> GetDataTransfer(const OrtMemoryInfo& mem_i
5757 return data_transfer;
5858 }
5959
60- if (mem_info.device .Type () == OrtDevice::GPU && mem_info.device .Vendor () == OrtDevice::NVIDIA) {
60+ if (mem_info.device .Type () == OrtDevice::GPU && mem_info.device .Vendor () == OrtDevice::VendorIds:: NVIDIA) {
6161#if defined(USE_CUDA) || defined(USE_CUDA_PROVIDER_INTERFACE)
6262 auto * cuda_provider_info = TryGetProviderInfo_CUDA ();
6363 if (cuda_provider_info != nullptr ) {
Original file line number Diff line number Diff line change @@ -218,7 +218,7 @@ TEST(LoraAdapterTest, VerifyDeviceCopy) {
218218 const auto & tensor_device = param.GetDeviceOrMapped ().Get <Tensor>();
219219 const auto & mem_info = tensor_device.Location ();
220220 ASSERT_EQ (mem_info.device .Type (), OrtDevice::GPU);
221- ASSERT_EQ (mem_info.device .Vendor (), OrtDevice::NVIDIA);
221+ ASSERT_EQ (mem_info.device .Vendor (), OrtDevice::VendorIds:: NVIDIA);
222222
223223 const auto & tensor_cpu = param.GetMapped ().Get <Tensor>();
224224 ASSERT_EQ (tensor_cpu.Shape ().Size (), tensor_device.Shape ().Size ());
You can’t perform that action at this time.
0 commit comments