Skip to content

Use dynamic_mode=off for theRock and fix unit tests#1027

Open
hsharsha wants to merge 2 commits into
rocm-jaxlib-v0.9.2from
rj_0.9.2_theRock_support
Open

Use dynamic_mode=off for theRock and fix unit tests#1027
hsharsha wants to merge 2 commits into
rocm-jaxlib-v0.9.2from
rj_0.9.2_theRock_support

Conversation

@hsharsha

@hsharsha hsharsha commented Jul 3, 2026

Copy link
Copy Markdown
Collaborator

Motivation

Technical Details

Test Plan

Test Result

Submission Checklist

Comment thread build_tools/rocm/rocm_xla.bazelrc Outdated
build:ci_single_gpu --flaky_test_attempts=3

build:ci_multi_gpu --action_env=XLA_FLAGS="--xla_gpu_force_compilation_parallelism=16 --xla_gpu_enable_llvm_module_compilation_parallelism=true"
build:ci_multi_gpu --action_env=XLA_FLAGS="--xla_gpu_force_compilation_parallelism=16"

Copy link
Copy Markdown
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

why don't keep --xla_gpu_enable_llvm_module_compilation_parallelism=true ?

Copy link
Copy Markdown
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

That flag didn't do nothing for us untill very recent XLA upstream.

@hsharsha hsharsha force-pushed the rj_0.9.2_theRock_support branch from 90aaaec to 8f0e438 Compare July 6, 2026 07:51
@hsharsha hsharsha force-pushed the rj_0.9.2_theRock_support branch from 8f0e438 to 83b1b16 Compare July 6, 2026 08:16
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants