mirror of
https://github.com/saymrwulf/onnxruntime.git
synced 2026-05-21 21:52:11 +00:00
29 lines
857 B
C++
29 lines
857 B
C++
// Copyright (c) Microsoft Corporation. All rights reserved.
|
|
// Licensed under the MIT License.
|
|
|
|
#pragma once
|
|
|
|
#include "fileHelpers.h"
|
|
#include "winrt_headers.h"
|
|
|
|
namespace APITest {
|
|
static void LoadModel(const std::wstring& modelPath,
|
|
winml::LearningModel& learningModel) {
|
|
std::wstring fullPath = FileHelpers::GetModulePath() + modelPath;
|
|
learningModel = winml::LearningModel::LoadFromFilePath(fullPath);
|
|
};
|
|
|
|
static uint64_t GetAdapterIdQuadPart(winml::LearningModelDevice& device) {
|
|
LARGE_INTEGER id;
|
|
id.LowPart = device.AdapterId().LowPart;
|
|
id.HighPart = device.AdapterId().HighPart;
|
|
return id.QuadPart;
|
|
};
|
|
|
|
static _LUID GetAdapterIdAsLUID(winml::LearningModelDevice& device) {
|
|
_LUID id;
|
|
id.LowPart = device.AdapterId().LowPart;
|
|
id.HighPart = device.AdapterId().HighPart;
|
|
return id;
|
|
}
|
|
}; // namespace APITest
|