tensorflow / tensorflow/java

How can I change the build config for the platform natives to mkl_aarch64

Open
#409 7 comments 0 reactions 0 assignees View on GitHub

Nobody has claimed this yet.

Dominant language
Java
Stars
928
Forks
227
PR merge metrics
No merged PRs in 30d

Description

System information

  • OS Platform and Distribution (e.g., Linux Ubuntu 16.04 x86_64): Aarch64 (Apple Sillicon)
  • TensorFlow installed from (source or binary): Source
  • TensorFlow version: 2.7
  • Java version (i.e., the output of java -version): 17
  • Java command line flags (e.g., GC parameters):
  • Installed from Maven Central?: No
  • Bazel version (if compiling from source): 3.7.2
  • GCC/Compiler version (if compiling from source): Apple clang version 13.0.0
  • CUDA/cuDNN version: N/A
  • GPU model and memory: Apple M1

When building core-api I can see from the logs it is targeting x86_64. I wish to target Aarch64.

./build.sh in core-api

Contributor guide

Open the contributing guide

First steps

  1. Read the whole issue, then the project's contributing guide.
  2. Comment on the issue to say you are picking it up — it saves two people doing the same work.
  3. Fork the repository and make your change on a branch.
  4. Open a pull request that references the issue number.

Research direction

Start with ./build.sh in core-api and inspect the build logs to find where the native platform is selected. Trace the configuration for the x86_64 target and determine what is needed for an Aarch64 or mkl_aarch64 build; done means core-api builds for the Apple M1 architecture.

Written by the indexing model from the issue text.

Assessment

Tech stack
java
Domain
build-system
Issue type
Feature
Difficulty
4/5
Estimated time
3-5 days
Activity status
Stale
Clarity
Needs clarification
Newbie friendliness
35/100

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.