|
10 | 10 | export ARCH_FLAGS="--allow_running_as_root"
|
11 | 11 | export DNNL_FLAGS="--use_dnnl"
|
12 | 12 | export OPENMP_FLAGS= # "--use_openmp"
|
| 13 | +export CUDAFLAGS="-v" |
13 | 14 | export CUDACXX="/usr/local/cuda/bin/nvcc"
|
14 | 15 | export CUDA_HOME="/usr/local/cuda"
|
15 | 16 | export CUDNN_HOME="/usr/local/cuda"
|
@@ -75,18 +76,24 @@ sedinplace '/WX/d' cmake/CMakeLists.txt
|
75 | 76 |
|
76 | 77 | # allow cross compilation for linux-arm64
|
77 | 78 | sedinplace 's/if args.arm or args.arm64:/if False:/g' tools/ci_build/build.py
|
| 79 | +sedinplace 's/is_linux() and platform.machine() == "x86_64"/False/g' tools/ci_build/build.py |
| 80 | +sedinplace 's/onnxruntime_target_platform STREQUAL "aarch64"/FALSE/g' cmake/CMakeLists.txt |
| 81 | +sedinplace 's/if (NOT APPLE)/if (FALSE)/g' cmake/onnxruntime_mlas.cmake |
| 82 | +sedinplace 's/!defined(__APPLE__)/0/g' onnxruntime/core/mlas/inc/mlas.h |
| 83 | +sedinplace 's/defined(__aarch64__) && defined(__linux__)/0/g' `find . -name *.cpp -o -name *.cc -o -name *.h` |
| 84 | +sedinplace 's/MLAS_CPUIDINFO::GetCPUIDInfo().HasArmNeon_I8MM()/false/g' onnxruntime/core/mlas/lib/platform.cpp |
78 | 85 |
|
79 | 86 | # work around toolchain issues on Mac and Windows
|
80 | 87 | patch -p1 < ../../../onnxruntime.patch
|
81 | 88 | #patch -p1 < ../../../onnxruntime-windows.patch # https://github.com/microsoft/onnxruntime/pull/7883
|
82 | 89 | sedinplace '/--Werror/d' cmake/CMakeLists.txt
|
83 | 90 | sedinplace "s/return 'ON'/return 'OFF'/g" tools/ci_build/build.py
|
84 | 91 | sedinplace "s/Visual Studio 1. 20../Ninja/g" tools/ci_build/build.py
|
85 |
| -sedinplace 's/Darwin|iOS/iOS/g' cmake/onnxruntime_providers.cmake |
86 |
| -sedinplace 's/-fvisibility=hidden//g' cmake/CMakeLists.txt cmake/adjust_global_compile_flags.cmake cmake/onnxruntime_providers.cmake |
87 |
| -sedinplace 's:/Yucuda_pch.h /FIcuda_pch.h::g' cmake/onnxruntime_providers.cmake |
88 |
| -sedinplace 's/${PROJECT_SOURCE_DIR}\/external\/cub//g' cmake/onnxruntime_providers.cmake |
89 |
| -sedinplace 's/ONNXRUNTIME_PROVIDERS_SHARED)/ONNXRUNTIME_PROVIDERS_SHARED onnxruntime_providers_shared)/g' cmake/onnxruntime_providers.cmake |
| 92 | +sedinplace 's/Darwin|iOS/iOS/g' cmake/onnxruntime_providers_cpu.cmake cmake/onnxruntime_providers.cmake |
| 93 | +sedinplace 's/-fvisibility=hidden//g' cmake/CMakeLists.txt cmake/adjust_global_compile_flags.cmake cmake/onnxruntime_providers_cpu.cmake cmake/onnxruntime_providers.cmake |
| 94 | +sedinplace 's:/Yucuda_pch.h /FIcuda_pch.h::g' cmake/onnxruntime_providers_cuda.cmake cmake/onnxruntime_providers.cmake |
| 95 | +sedinplace 's/${PROJECT_SOURCE_DIR}\/external\/cub//g' cmake/onnxruntime_providers_cuda.cmake cmake/onnxruntime_providers.cmake |
| 96 | +sedinplace 's/ONNXRUNTIME_PROVIDERS_SHARED)/ONNXRUNTIME_PROVIDERS_SHARED onnxruntime_providers_shared)/g' cmake/onnxruntime_providers_cpu.cmake cmake/onnxruntime_providers.cmake |
90 | 97 | sedinplace 's/DNNL_TAG v.*)/DNNL_TAG v3.4.1)/g' cmake/external/dnnl.cmake
|
91 | 98 | sedinplace 's/DNNL_SHARED_LIB libdnnl.1.dylib/DNNL_SHARED_LIB libdnnl.2.dylib/g' cmake/external/dnnl.cmake
|
92 | 99 | sedinplace 's/DNNL_SHARED_LIB libdnnl.so.1/DNNL_SHARED_LIB libdnnl.so.2/g' cmake/external/dnnl.cmake
|
|
0 commit comments