From 6e16164329d78c60742cdcb3c4ea0cea1aa0112d Mon Sep 17 00:00:00 2001 From: Valery Chernov Date: Thu, 20 Jan 2022 17:20:45 +0300 Subject: [PATCH] small fixes for correct CI tests --- java/src/main/java/ai/onnxruntime/OrtSession.java | 3 +-- onnxruntime/test/nuphar_tvm/tvm_basic_test.cc | 2 +- onnxruntime/test/nuphar_tvm/tvm_demo/demo_compiler.cc | 2 +- 3 files changed, 3 insertions(+), 4 deletions(-) diff --git a/java/src/main/java/ai/onnxruntime/OrtSession.java b/java/src/main/java/ai/onnxruntime/OrtSession.java index 34b707f3de9f2..6dcaf8e451abc 100644 --- a/java/src/main/java/ai/onnxruntime/OrtSession.java +++ b/java/src/main/java/ai/onnxruntime/OrtSession.java @@ -1045,8 +1045,7 @@ private native void addNuphar( long apiHandle, long nativeHandle, int allowUnalignedBuffers, String settings) throws OrtException; - private native void addTvm( - long apiHandle, long nativeHandle, String settings) + private native void addTvm(long apiHandle, long nativeHandle, String settings) throws OrtException; private native void addDirectML(long apiHandle, long nativeHandle, int deviceId) diff --git a/onnxruntime/test/nuphar_tvm/tvm_basic_test.cc b/onnxruntime/test/nuphar_tvm/tvm_basic_test.cc index dddc2e6e04209..6fa1cf3b250c1 100644 --- a/onnxruntime/test/nuphar_tvm/tvm_basic_test.cc +++ b/onnxruntime/test/nuphar_tvm/tvm_basic_test.cc @@ -13,7 +13,7 @@ #include "core/session/onnxruntime_cxx_api.h" #include "test/framework/test_utils.h" #include "test/test_environment.h" -#include "test/tvm/tvm_demo/demo_compiler.h" +#include "test/nuphar_tvm/tvm_demo/demo_compiler.h" #include diff --git a/onnxruntime/test/nuphar_tvm/tvm_demo/demo_compiler.cc b/onnxruntime/test/nuphar_tvm/tvm_demo/demo_compiler.cc index 0f0685cd14313..ff2f9076a794d 100644 --- a/onnxruntime/test/nuphar_tvm/tvm_demo/demo_compiler.cc +++ b/onnxruntime/test/nuphar_tvm/tvm_demo/demo_compiler.cc @@ -1,7 +1,7 @@ // Copyright (c) Microsoft Corporation. All rights reserved. // Licensed under the MIT License. -#include "test/tvm/tvm_demo/demo_compiler.h" +#include "test/nuphar_tvm/tvm_demo/demo_compiler.h" #include "core/codegen/passes/scheduler/schedule_utils.h" #include "core/codegen/passes/utils/ort_tvm_utils.h"