Skip to content

Commit

Permalink
small fixes for correct CI tests
Browse files Browse the repository at this point in the history
  • Loading branch information
Valery Chernov committed Jan 20, 2022
1 parent ed8d6cf commit 6e16164
Show file tree
Hide file tree
Showing 3 changed files with 3 additions and 4 deletions.
3 changes: 1 addition & 2 deletions java/src/main/java/ai/onnxruntime/OrtSession.java
Original file line number Diff line number Diff line change
Expand Up @@ -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)
Expand Down
2 changes: 1 addition & 1 deletion onnxruntime/test/nuphar_tvm/tvm_basic_test.cc
Original file line number Diff line number Diff line change
Expand Up @@ -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 <tvm/runtime/ndarray.h>

Expand Down
2 changes: 1 addition & 1 deletion onnxruntime/test/nuphar_tvm/tvm_demo/demo_compiler.cc
Original file line number Diff line number Diff line change
@@ -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"
Expand Down

0 comments on commit 6e16164

Please sign in to comment.