mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-14 10:56:06 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add @xla to Jax's platform mappings config

Browse Source 
PiperOrigin-RevId: 620155135
This commit is contained in:
Kanglan Tang 2024-03-28 22:14:21 -07:00 committed by jax authors
parent 84156f359f
commit 1f0ffe6a66
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
platform_mappings
View File

@ -1,11 +1,11 @@
platforms: platforms:
# Maps "--platforms=//tools/toolchains/cross_compile/config:darwin_x86_64" # Maps "--platforms=//tools/toolchains/cross_compile/config:darwin_x86_64"
# to "--cpu=darwin". # to "--cpu=darwin".
//tools/toolchains/cross_compile/config:darwin_x86_64 @xla//tools/toolchains/cross_compile/config:darwin_x86_64
--cpu=darwin --cpu=darwin
flags: flags:
# Maps "--cpu=darwin" to # Maps "--cpu=darwin" to
# "--platforms=//tools/toolchains/cross_compile/config:darwin_x86_64". # "--platforms=//tools/toolchains/cross_compile/config:darwin_x86_64".
--cpu=darwin --cpu=darwin
//tools/toolchains/cross_compile/config:darwin_x86_64 @xla//tools/toolchains/cross_compile/config:darwin_x86_64