ROCm/jax

Pull Requests Commits

Adjust Indentation

AratiGanesh committed 81 days ago

d0af271e

Merge branch 'rocm-jaxlib-v0.6.0' into arati/fix_hipblas_resource_issue

AratiGanesh committed 82 days ago

Verified da4319ec

Move HIP inline functions to anonymous namespaces

AratiGanesh committed 86 days ago

62a87911

[0.6.0-UT] Adding third field to return value of get_rocm_version (#542)

gulsumgudukbay committed 94 days ago

Verified 9fc59609

[0.6.0-UT] Adding python equivalent of run_multi_gpu.sh (#543)

gulsumgudukbay committed 95 days ago

Verified 4528432b

[0.6.0-UT] Unskipping passing tests (#540)

gulsumgudukbay committed 95 days ago

Verified 4b3d54ab

Fix GPU handle pool singleton aliasing for HipSPARSE (#535)

AratiGanesh committed 95 days ago

Verified f1aaf9a9

Revert "[0.6.0-UT] Skipping aborted tests (HIP runtime issue)" (#539)

gulsumgudukbay committed 96 days ago

Verified 1a647fa6

Revert "Remove old build scripts (#532)" (#536)

charleshofer committed 100 days ago

Verified db216443

Replace GPU library wrapper macros with type-safe inline functions

AratiGanesh committed 102 days ago

58e32929

AratiGanesh committed 104 days ago

ab991b41

Add opaque pointer typedef for hipsparseHandle_t to prevent singleton aliasing

AratiGanesh committed 104 days ago

756ead61

Remove old build scripts (#532)

charleshofer committed 105 days ago

Verified d9b4b407

[0.6.0-UT] skipped more tests for build 33 (#531)

gulsumgudukbay committed 127 days ago

Verified 4ee8db98

[0.6.0-UT] Skipping aborted tests (HIP runtime issue) (#530)

gulsumgudukbay committed 129 days ago

Verified c07c3a15

[0.6.0-UT] Adding abort support to run_multi_gpu (#529)

gulsumgudukbay committed 129 days ago

Verified 68d7228c

[0.6.0-UT] Transferring Multi-GPU tests (#527)

gulsumgudukbay committed 129 days ago

Verified f697dde0

unskip pallas tests that were skipped due to LLVm error

gulsumgudukbay committed 129 days ago

41a32e35

addressing Charlie's comments

gulsumgudukbay committed 136 days ago

248cf432

removed a print stmt

gulsumgudukbay committed 136 days ago

e1236c74

Adding abort test info in run_single_gpu.py logs

gulsumgudukbay committed 136 days ago

1ff07701

fix is_device_rocm function call

root committed 139 days ago

bcc84983

Fix GPU handle pool singleton aliasing (#525)

phambinhfin committed 143 days ago

Verified 9890a51d

Skip test_dot_precision() with TF32 for Navi architecture (#520)

Arech8 committed 144 days ago

Verified 2c4c7064

adding additional skips only for MI350 for LLVM cannot scalarize ERROR

root committed 144 days ago

e5fc666b

Bugfix bazel debug rule (#511)

Arech8 committed 146 days ago

Verified 9c088fc9

The reshape call has a numerical bug where the target reshape size is always higher than array size if num_devices is not a nonnegative power of two. To fix this, this commit adds padding

gulsumgudukbay committed 158 days ago

3d6b521d

Fix test_gpu_distributed_initialize() to get a more proper GPUs number (#503)

Arech8 committed 164 days ago

Verified 03684216

Fix test_gpu_distributed_initialize() to take number of available GPUs into account

Arech8 committed 165 days ago

58a5c560

Revert "unskip test_select_n"

gulsumgudukbay committed 165 days ago

81ba3269

Older