Search results

Reduce block sizes for GPUs with less shared memory

CI Optional GPU PresubmitLabel to flag PR to run additional GPU testing not in standard presubmitsLabel to flag PR to run additional GPU testing not in standard presubmits

kokoro:force-run

pull readyReady for copybara import and testingReady for copybara import and testing

Status: Merged (completed).

jax-ml/jaxnumber 32072

#32072 In jax-ml/jax;

· by olupton was closed on Sep 26, 2025

Test dumping HLO when reading compilation cache

kokoro:force-run

pull readyReady for copybara import and testingReady for copybara import and testing

Status: Open (in progress).

jax-ml/jaxnumber 31328

#31328 In jax-ml/jax;

· olupton opened on Aug 26, 2025

RTX PRO 6000 Blackwell: skip more tests that use too much smem

kokoro:force-run

pull readyReady for copybara import and testingReady for copybara import and testing

Status: Merged (completed).

jax-ml/jaxnumber 31183

#31183 In jax-ml/jax;

· by olupton was closed on Aug 20, 2025

fused_attention_stablehlo_test_gpu: test skip for 12.0

kokoro:force-run

pull readyReady for copybara import and testingReady for copybara import and testing

Status: Merged (completed).

jax-ml/jaxnumber 30272

#30272 In jax-ml/jax;

· by olupton was closed on Jul 17, 2025

test_multiprocess: only initialize GPUs that exist

kokoro:force-run

pull readyReady for copybara import and testingReady for copybara import and testing

Status: Merged (completed).

jax-ml/jaxnumber 30248

#30248 In jax-ml/jax;

· by olupton was closed on Jul 21, 2025

scaled_matmul_stablehlo_test: add multiaccelerator tag

kokoro:force-run

pull readyReady for copybara import and testingReady for copybara import and testing

Status: Merged (completed).

jax-ml/jaxnumber 30047

#30047 In jax-ml/jax;

· by olupton was closed on Jul 8, 2025

cuFFT: remove deprecated enum values

kokoro:force-run

pull readyReady for copybara import and testingReady for copybara import and testing

Status: Merged (completed).

jax-ml/jaxnumber 29739

#29739 In jax-ml/jax;

· by olupton was closed on Jun 25, 2025

//tests:scaled_matmul_stablehlo_test: fix for xla#27096

kokoro:force-run

pull readyReady for copybara import and testingReady for copybara import and testing

Status: Merged (completed).

jax-ml/jaxnumber 29294

#29294 In jax-ml/jax;

· by olupton was closed on Jun 11, 2025

Parametrize build system on CUDA major version

kokoro:force-run

pull readyReady for copybara import and testingReady for copybara import and testing

Status: Merged (completed).

jax-ml/jaxnumber 28968

#28968 In jax-ml/jax;

· by olupton was closed on Jul 21, 2025

Add extra cuBLAS/cuDNN version checks

kokoro:force-run

pull readyReady for copybara import and testingReady for copybara import and testing

Status: Merged (completed).

jax-ml/jaxnumber 28931

#28931 In jax-ml/jax;

· by olupton was closed on Jun 13, 2025

Revert "jax-cuda12-plugin: require nvidia-cublas-cu12<12.9"

kokoro:force-run

pull readyReady for copybara import and testingReady for copybara import and testing

Status: Merged (completed).

jax-ml/jaxnumber 28897

#28897 In jax-ml/jax;

· by olupton was closed on May 21, 2025

pytest: use importlib mode by default

kokoro:force-run

pull readyReady for copybara import and testingReady for copybara import and testing

Status: Merged (completed).

jax-ml/jaxnumber 28650

#28650 In jax-ml/jax;

· by olupton was closed on May 9, 2025