Skip to content

Commit 5580d43

Browse files
yuslepukhinCopilot
andcommitted
Update onnxruntime/test/lora/lora_test.cc
Co-authored-by: Copilot <[email protected]>
1 parent ab442e1 commit 5580d43

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

onnxruntime/core/session/lora_adapters.cc

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff 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) {

onnxruntime/test/lora/lora_test.cc

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff 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());

0 commit comments

Comments
 (0)