jax
Adding gpu_device_targets parameter
#233
Merged

Adding gpu_device_targets parameter #233

srayasam-amd merged 26 commits into rocm-jaxlib-v0.4.35-qa from srayasam_test
srayasam-amd
srayasam-amd Update Dockerfile.manylinux_2_28_x86_64.rocm
a3d34818
srayasam-amd Update ci_build.sh
f72eb95f
srayasam-amd Update build_wheels.py
460b3c35
srayasam-amd Update ci_build
5da1c4e8
srayasam-amd Update ci_build.sh
a53822fa
srayasam-amd Update ci_build.sh
50013656
srayasam-amd Update ci_build.sh
3403473e
srayasam-amd Update ci_build.sh
58f2c0e4
srayasam-amd Update Dockerfile.manylinux_2_28_x86_64.rocm
3ba06823
srayasam-amd Update Dockerfile.manylinux_2_28_x86_64.rocm
d917a225
srayasam-amd Update Dockerfile.manylinux_2_28_x86_64.rocm
bdea57c4
srayasam-amd Update ci_build.sh
6d50f2a2
srayasam-amd Update ci_build.sh
3e64d0b6
srayasam-amd Update Dockerfile.manylinux_2_28_x86_64.rocm
6f8b984b
srayasam-amd Update ci_build.sh
d3cdd4f7
srayasam-amd Update ci_build.sh
082d0575
srayasam-amd Update ci_build.sh
96315972
srayasam-amd Update Dockerfile.manylinux_2_28_x86_64.rocm
2aa8e23b
srayasam-amd Update ci_build
90f95a2e
srayasam-amd Update build_wheels.py
c95d2676
srayasam-amd Update ci_build
ae9f449c
srayasam-amd Update ci_build
40a99379
srayasam-amd Update ci_build.sh
aba4376e
srayasam-amd Update ci_build
53c845d3
srayasam-amd Update build_wheels.py
efa7bf4f
srayasam-amd Update ci_build
35ffecdc
srayasam-amd srayasam-amd changed the title Srayasam test Adding gpu_device_targets parameter 309 days ago
kiran-thumma
kiran-thumma approved these changes on 2025-02-20
srayasam-amd srayasam-amd merged f21c8260 into rocm-jaxlib-v0.4.35-qa 306 days ago

Login to write a write a comment.

Login via GitHub

Reviewers
Assignees
No one assigned
Labels
Milestone