Rocm jaxlib v0.4.30 qa nccl maxnchannels#75

Closed

hsharsha wants to merge 46 commits intorocm-jaxlib-v0.4.30from rocm-jaxlib-v0.4.30-qa_nccl_maxnchannels

+1,476-640

Commits on Jul 11, 2024

[ROCm]: Fix LLVM path issue for ROCm 6.3
Rahul Batra
committed

Commits on Jul 25, 2024

[ROCm]: Fix LLVM path for ROCm 6.2
Rahul Batra
committed

Commits on Aug 5, 2024

Commits on Aug 8, 2024

Let the other stream wait for the main stream before issuing memcpy d2h

zhenying-liu
authored and
i-chaochen
committed
Merge pull request #34 from ROCm/rocm-jaxlib-v0.4.30-qa-d2hmem-stream-copy
i-chaochen
authored

Commits on Aug 16, 2024

workspace fixing

pemeliya
authored and
hsharsha
committed
[ROCM] Updated fp8 matmul with adjustments for updated hipBlasLt support

wenchenvincent
authored and
hsharsha
committed
[ROCM] Addressed reviewer comment.

wenchenvincent
authored and
hsharsha
committed
[ROCM] Fix build after 21311f2

draganmladjenovic
authored and
hsharsha
committed
[ROCM] Add basic scaffolding and enable MLIR fusion

draganmladjenovic
authored and
hsharsha
committed
Enable dot algorithms for AMD GPUs
hsharsha
committed

Commits on Aug 23, 2024

added bias pointer workaround

pemeliya
authored and
hsharsha
committed
added precision settings for autotuner and buffer_comparator small refactoring, added verbose flag

pemeliya
authored and
hsharsha
committed
added precision settings for autotuner and buffer_comparator small refactoring, added verbose flag

pemeliya
authored and
hsharsha
committed
small changes

i-chaochen
authored and
hsharsha
committed
adopted changes

i-chaochen
authored and
hsharsha
committed

Commits on Aug 29, 2024

Commits on Sep 16, 2024

scrub navi
ScXfjiang
committed

Commits on Sep 17, 2024

Merge pull request #42 from ROCm/rocm-jaxlib-v0.4.30-qa_navi_scrub
ScXfjiang
authored

Commits on Sep 20, 2024

PR openxla#16938 : Add NANOO FP8 support for collaborative communication unit tests
ScXfjiang
committed

Commits on Sep 24, 2024

Merge pull request #45 from ROCm/rocm-jaxlib-v0.4.30-qa_collective_fp8
ScXfjiang
authored

Commits on Sep 25, 2024

Commits on Oct 2, 2024

Reset stream function in Gemm algorithm picker (#39 )

hsharsha
and
pemeliya
authored

Commits on Oct 9, 2024

[ROCm] Fixed linker issues with rocblas_get_version_string_size and rocblas_get_version_string (#52 )
zoranjovanovic-ns
authored

Commits on Oct 11, 2024

Add multigpu script and disable triton tests
Harsha HS
committed

Commits on Oct 15, 2024

Merge pull request #53 from ROCm/rocm-jaxlib-v0.4.30-qa-multigpu-disable-triton
hsharsha
authored

Commits on Oct 22, 2024

Commits on Oct 28, 2024

buffer init fix and gpu_hlo_runner test
pemeliya
committed

Commits on Oct 30, 2024

Merge pull request #59 from ROCm/r0.4.30_buffer_init_and_hlo_runner
pemeliya
authored

Commits on Nov 13, 2024

Commits on Nov 19, 2024

[ROCm] Pass AMDGPU_TARGETS to crosstool wrapper

hsharsha
authored and
Ruturaj4
committed
[Rocm] fix arch
Ruturaj4
committed
Merge pull request #63 from ROCm/ci_clang_31_2

pramenku
authored and
Ruturaj4
committed

Commits on Nov 20, 2024

Skip gpu_hlo_runner_test if input is not provided (#71 )
mmakevic-amd
authored

Commits on Nov 28, 2024

Add NCCL_MAX_NCHANNELS env variable to multi gpu tests
hsharsha
committed