Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Linux arm64へのビルドを追加 #13

Merged
merged 5 commits into from
Dec 24, 2022
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
6 changes: 6 additions & 0 deletions .github/actions/auto_gen_bind_pr/action.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -14,6 +14,12 @@ runs:
shell: bash
run: echo "##[set-output name=branch;]$(echo ${GITHUB_REF#refs/heads/})"
id: extract_branch
- name: Install cross compiler for aarch64
if: inputs.triple == 'aarch64-unknown-linux-gnu'
shell: bash
run: |
sudo apt update
sudo apt install gcc-aarch64-linux-gnu g++-aarch64-linux-gnu
- uses: actions-rs/toolchain@v1
with:
toolchain: stable
Expand Down
2 changes: 2 additions & 0 deletions .github/workflows/gen_bind.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -13,6 +13,8 @@ jobs:
include:
- os: ubuntu-latest
triple: x86_64-unknown-linux-gnu
- os: ubuntu-latest
triple: aarch64-unknown-linux-gnu
- os: windows-latest
triple: x86_64-pc-windows-msvc
- os: windows-latest
Expand Down
3 changes: 2 additions & 1 deletion onnxruntime-sys/build.rs
Original file line number Diff line number Diff line change
Expand Up @@ -435,7 +435,8 @@ impl OnnxPrebuiltArchive for Triplet {
Cow::from(format!("{}-{}", self.os.as_onnx_str(), "x64"))
}
(Os::Windows, Architecture::X86, Accelerator::None)
| (Os::MacOs, Architecture::X86_64, Accelerator::None) => Cow::from(format!(
| (Os::MacOs, Architecture::X86_64, Accelerator::None)
| (Os::Linux, Architecture::Arm64, Accelerator::None) => Cow::from(format!(
"{}-{}",
self.os.as_onnx_str(),
self.arch.as_onnx_str()
Expand Down
6 changes: 6 additions & 0 deletions onnxruntime-sys/src/generated/bindings.rs
Original file line number Diff line number Diff line change
Expand Up @@ -4,6 +4,12 @@ include!(concat!(
"/src/generated/linux/x86_64/bindings.rs"
));

#[cfg(all(target_os = "linux", target_arch = "aarch64"))]
include!(concat!(
env!("CARGO_MANIFEST_DIR"),
"/src/generated/linux/aarch64/bindings.rs"
));

#[cfg(all(target_os = "macos", target_arch = "x86_64"))]
include!(concat!(
env!("CARGO_MANIFEST_DIR"),
Expand Down
Loading