From 4471fcd32e27602f67e221d224514320152ddeaf Mon Sep 17 00:00:00 2001 From: Weiming Zhao Date: Wed, 27 Oct 2021 23:03:21 -0700 Subject: [PATCH 1/2] [ODLA/TensorRT] Add arch 80 --- ODLA/platforms/tensorrt/plugins/CMakeLists.txt | 1 + 1 file changed, 1 insertion(+) diff --git a/ODLA/platforms/tensorrt/plugins/CMakeLists.txt b/ODLA/platforms/tensorrt/plugins/CMakeLists.txt index e33315918..8b25cca6c 100644 --- a/ODLA/platforms/tensorrt/plugins/CMakeLists.txt +++ b/ODLA/platforms/tensorrt/plugins/CMakeLists.txt @@ -26,6 +26,7 @@ else() 61 70 75 + 80 ) message(STATUS "GPU_ARCHS is not defined. Generating CUDA code for default SMs: ${GPU_ARCHS}") endif() From 4dc03b2e1c4a63a6acecdade6012f016a0f266aa Mon Sep 17 00:00:00 2001 From: Weiming Zhao Date: Thu, 28 Oct 2021 14:30:35 -0700 Subject: [PATCH 2/2] try --- tests/unittests/lit_cases/test_tensorrt/test_and2d_tensorrt.cc | 2 +- tests/unittests/lit_cases/test_tensorrt/test_and3d_tensorrt.cc | 2 +- tests/unittests/lit_cases/test_tensorrt/test_not_3d_tensorrt.cc | 2 +- 3 files changed, 3 insertions(+), 3 deletions(-) diff --git a/tests/unittests/lit_cases/test_tensorrt/test_and2d_tensorrt.cc b/tests/unittests/lit_cases/test_tensorrt/test_and2d_tensorrt.cc index 395d9dd93..739338db7 100644 --- a/tests/unittests/lit_cases/test_tensorrt/test_and2d_tensorrt.cc +++ b/tests/unittests/lit_cases/test_tensorrt/test_and2d_tensorrt.cc @@ -23,7 +23,7 @@ // RUN: %halo_compiler -target cxx -batch-size 1 %halo_compile_flags %data_path/test_and2d/model.onnx -o %t.cc // RUN: %cxx -c -fPIC -o %t.o %t.cc -I%odla_path/include // RUN: %cxx -g %s %t.o %t.bin -I%T -I%odla_path/include -I%unittests_path -I%data_path/test_and2d/test_data_set_0 %odla_link %device_link -lodla_tensorrt -o %t_tensorrt.exe -Wno-deprecated-declarations -// RUN: ODLA_TRT_USE_EXPLICIT_BATCH=1 %t_tensorrt.exe 0.0001 0 tensorrt %data_path/test_and2d | FileCheck %s +// RUN: %t_tensorrt.exe 0.0001 0 tensorrt %data_path/test_and2d | FileCheck %s // CHECK: Result Pass // clang-format on diff --git a/tests/unittests/lit_cases/test_tensorrt/test_and3d_tensorrt.cc b/tests/unittests/lit_cases/test_tensorrt/test_and3d_tensorrt.cc index c4e0573c6..0ed790722 100644 --- a/tests/unittests/lit_cases/test_tensorrt/test_and3d_tensorrt.cc +++ b/tests/unittests/lit_cases/test_tensorrt/test_and3d_tensorrt.cc @@ -23,7 +23,7 @@ // RUN: %halo_compiler -target cxx -batch-size 1 %halo_compile_flags %data_path/test_and3d/model.onnx -o %t.cc // RUN: %cxx -c -fPIC -o %t.o %t.cc -I%odla_path/include // RUN: %cxx -g %s %t.o %t.bin -I%T -I%odla_path/include -I%unittests_path -I%data_path/test_and3d/test_data_set_0 %odla_link %device_link -lodla_tensorrt -o %t_tensorrt.exe -Wno-deprecated-declarations -// RUN: ODLA_TRT_USE_EXPLICIT_BATCH=1 %t_tensorrt.exe 0.0001 0 tensorrt %data_path/test_and3d | FileCheck %s +// RUN: %t_tensorrt.exe 0.0001 0 tensorrt %data_path/test_and3d | FileCheck %s // CHECK: Result Pass // clang-format on diff --git a/tests/unittests/lit_cases/test_tensorrt/test_not_3d_tensorrt.cc b/tests/unittests/lit_cases/test_tensorrt/test_not_3d_tensorrt.cc index c5f78e718..87cdf3590 100644 --- a/tests/unittests/lit_cases/test_tensorrt/test_not_3d_tensorrt.cc +++ b/tests/unittests/lit_cases/test_tensorrt/test_not_3d_tensorrt.cc @@ -22,7 +22,7 @@ // RUN: %halo_compiler -target cxx -batch-size 1 %halo_compile_flags %data_path/test_not_3d/model.onnx -o %t.cc // RUN: %cxx -c -fPIC -o %t.o %t.cc -I%odla_path/include // RUN: %cxx -g %s %t.o %t.bin -I%T -I%odla_path/include -I%unittests_path -I%data_path/test_not_3d/test_data_set_0 %odla_link %device_link -lodla_tensorrt -o %t_tensorrt.exe -Wno-deprecated-declarations -// RUN: ODLA_TRT_USE_EXPLICIT_BATCH=1 %t_tensorrt.exe 0.0001 0 tensorrt %data_path/test_not_3d | FileCheck %s +// RUN: %t_tensorrt.exe 0.0001 0 tensorrt %data_path/test_not_3d | FileCheck %s // CHECK: Result Pass // clang-format on