Ruturaj4
ce7347a52b
Fix circular import in pallas core file
2025-04-09 19:57:24 -05:00
gulsumgudukbay
ac21549df0
pallas/ops_test health check suppressed for test_select_n
2025-04-09 16:26:22 -05:00
gulsumgudukbay
53f9f29368
Having ignore warning decorator before the parameterized test decorator gives an error that tells the ignorewarning decorator hinders the parameterized test creation, hence no tests are ran for testToeplitzConstruction. The extra decorator is removed as Ruturaj created an ignorewarning statement in a finer granularity.
2025-04-09 08:41:56 -05:00
Ruturaj Vaidya
75fc650d0b
Fix ignore future warnings for scipy ( #348 )
2025-04-08 13:25:45 -05:00
Ruturaj4
f79cd50002
Ignore future scipy warning and not treat it as an error
2025-04-08 11:10:01 -05:00
Jake VanderPlas
fd0366b169
Skip vecmat & matvec in NumPy tests.
...
(cherry picked from commit f182aa8eddbdb99d2299f2eb2153b03078a124da)
2025-04-07 16:39:02 -05:00
Mathew Odden
44c9b9afd0
Add gfx1101 to build targets ( #332 )
2025-04-03 11:51:17 -05:00
Mathew Odden
8bfcf73c07
Add canonicalization of python versions
...
The ci_build parsing will now drop the
revision portion of the requested python
version targets to build for, since users
were passing in old buggy versions.
Using only major.minor will result in the latest
of the minor series being used, and won't affect
any build outputs like bytecode or ABI bindings.
Those are tied to major.minor versions of CPython.
(cherry picked from commit 5bf08753d6da9ef46d2c6b5388dfa68952212816)
2025-03-31 13:37:34 -05:00
Mathew Odden
0a646331ac
Port gpu_device_targets changes from rocm-main
2025-03-31 13:37:34 -05:00
Mathew Odden
0a527a2d0b
Split parse_gpu_targets test cases
...
(cherry picked from commit 5024cf8aef13ae049276093716621343b0e41742)
2025-03-31 13:37:34 -05:00
Ruturaj Vaidya
4adc89b74e
Don't skip fp8 tests on 35-qa branch as they all are passing ( #307 )
2025-03-21 23:06:58 -05:00
Mathew Odden
ef1d56184b
Fix missing defaults for GPU target arch'es ( #293 )
...
Move defaults for these up to ci_build python
We also add some error catching logic in the
python script to clean up weirdness coming
from the build scripts.
Also clean up some formatting and typos
2025-03-17 17:15:53 -05:00
Mathew Odden
792fceeafe
Fix gpu targets arg issue ( #291 )
...
nargs was eating the last positional arg,
so change this to a comma separated string list.
Also make arg syntax consistent with others.
2025-03-17 14:48:13 -05:00
Mathew Odden
993adb4944
Fix auditwheel version issue ( #289 )
...
Auditwheel 6.3.0 changed/removed the lddtree function
so cap constraint to 6.2.x
(cherry picked from commit 57e77ce3a82d93848ccba557b007084605b40f22)
2025-03-17 13:03:56 -05:00
Ruturaj Vaidya
0c94158021
Update commit hash ( #287 )
2025-03-17 12:11:07 -05:00
Ruturaj Vaidya
8c48c0f62a
35 several uts ( #282 )
...
* Don't perform size 0 slices into scipy rotations.
This is disallowed by scipy after https://github.com/scipy/scipy/pull/21776 .
(cherry picked from commit 834e71bbe1e4e23de8f658e7380c385be7c5099a)
* test: change random matrix generation for Rotation
(cherry picked from commit 4d433d063e2274d2998913d615aa168889a91b9a)
* Skip lax_scipy_special_functions_test as it will be fixed in the next JAX version
---------
Co-authored-by: Peter Hawkins <phawkins@google.com>
Co-authored-by: Jake VanderPlas <jakevdp@google.com>
2025-03-14 09:41:59 -05:00
Ruturaj Vaidya
9e785a071b
Fix extend test for ROCm ( #279 )
2025-03-13 19:14:04 -05:00
Ruturaj Vaidya
b5c700c9a7
Temporarily allow deprecation warnings for scipy.special.lpmn and scipy.special.sph_harm. ( #278 )
...
These functions are deprecated in scipy 1.15.0. I'll fix this properly soon, but let's start by getting CI working again!
PiperOrigin-RevId: 712512363
Co-authored-by: Dan Foreman-Mackey <danfm@google.com>
2025-03-13 16:16:19 -05:00
Ruturaj Vaidya
d90651ba32
Use HIPBLAS_V2 ( #222 ) ( #277 )
...
Co-authored-by: Harsha HS <Harsha.HavanurShamsundara@amd.com>
2025-03-12 15:02:42 -05:00
Dragan Mladjenovic
a8c11ba79e
[pallas:triton] Fix atomic min/max lowering for unsigned integers and float types ( #263 )
2025-03-10 10:39:22 -05:00
JD
01c8d7feb8
remove merge artifact leftovers ( #267 )
2025-03-07 11:51:23 -06:00
Zahid Iqbal
661e0e8fef
removing csv result compilation after Unit test... ( #260 )
2025-03-05 13:37:39 -06:00
JD
48f8e59399
add gfx1101 target ( #252 )
2025-03-05 11:43:49 -06:00
Ruturaj Vaidya
fd1e518e44
Fix RNN patch for 35-qa branch ( #245 )
2025-02-27 10:03:47 -06:00
Ruturaj Vaidya
c5ee54f979
Fix bytes_limit key of mem_stats ( #243 )
2025-02-26 10:17:54 -06:00
Ruturaj Vaidya
bbaaec088c
Skip tests on ROCm ( #235 ) ( #242 )
2025-02-26 09:16:02 -06:00
srayasam-amd
f21c826035
Adding gpu_device_targets parameter ( #233 )
...
* Update Dockerfile.manylinux_2_28_x86_64.rocm
adding gpu_device_targets
* Update ci_build.sh
* Update build_wheels.py
* Update ci_build
* Update ci_build.sh
adding if condition for gpu_Device_targets
* Update ci_build.sh
* Update ci_build.sh
* Update ci_build.sh
* Update Dockerfile.manylinux_2_28_x86_64.rocm
* Update Dockerfile.manylinux_2_28_x86_64.rocm
* Update Dockerfile.manylinux_2_28_x86_64.rocm
* Update ci_build.sh
* Update ci_build.sh
* Update Dockerfile.manylinux_2_28_x86_64.rocm
* Update ci_build.sh
* Update ci_build.sh
* Update ci_build.sh
* Update Dockerfile.manylinux_2_28_x86_64.rocm
* Update ci_build
* Update build_wheels.py
* Update ci_build
* Update ci_build
* Update ci_build.sh
* Update ci_build
* Update build_wheels.py
* Update ci_build
2025-02-20 10:34:40 +05:30
JD
a844a922df
Silence C23 warning, causing error in upb target for XLA ( #230 )
...
* Silence C23 warning, creating error in upb target for XLA
* Copy CI workfow from rocm-main
* Add option to ci_build to run different tests
* Fix tty
---------
Co-authored-by: Charles Hofer <Charles.Hofer@amd.com>
2025-02-13 12:21:23 -06:00
Ruturaj Vaidya
39f7db687e
Change jaxlib/rocm/BUILD nanobind_extension to pybind ( #226 )
2025-02-07 16:07:02 -06:00
Ruturaj Vaidya
dceb5310fe
[ROCm] Implement RNN support ( #217 )
2025-02-07 11:08:12 -06:00
Ruturaj Vaidya
da40f36d00
Fix run_multi_gpu script multi-gpu issue and refactor code ( #208 )
2025-01-19 18:03:00 -06:00
Ruturaj Vaidya
f99fbcb3c5
Add gfx12xx archs ( #205 )
2025-01-14 10:52:00 -06:00
Kiran Thumma
5d56aaf1e3
Merge pull request #201 from ROCm/rocm-jaxlib-v0.4.35-qa-fix2
...
buildkit option for docker build
2025-01-13 06:43:44 -06:00
kithumma
f075f904b2
buildkit option for docker build
2025-01-11 06:09:27 +00:00
Kiran Thumma
c5e32ac558
Merge pull request #195 from ROCm/rocm-jaxlib-v0.4.35-qa-fix
...
fix cdn
2025-01-08 06:12:51 -06:00
kithumma
590ac67c4b
fix cdn
2024-12-30 17:11:51 +00:00
Ruturaj Vaidya
8985f20e4f
Merge pull request #169 from ROCm/ci_qa_sync
...
[JAX] Ci qa sync
2024-12-02 10:28:21 -06:00
Ruturaj Vaidya
fab53b06f9
[ROCm] Update XLA commit hash ( #168 )
2024-12-02 06:51:34 -06:00
Ruturaj Vaidya
20362f83c6
[ROCm] Remove exit on error ( #167 )
2024-12-02 06:51:25 -06:00
Ruturaj4
010aea0c56
Add argument to override base docker in dockerfile
2024-12-02 06:49:40 -06:00
Ruturaj4
3b3083bbc4
[ROCm] ci build and dockerfile changes
2024-12-02 06:49:32 -06:00
Ruturaj4
b8d2dedeed
[ROCm] Change run_multi_gpu set opts
2024-12-02 06:49:22 -06:00
Ruturaj4
a5a1417587
BUILD file fixes and linter run
2024-12-02 06:49:15 -06:00
Ruturaj4
addca5d517
[ROCm] fix jax and wheelhouse relative paths
2024-12-02 06:49:10 -06:00
Ruturaj4
39c0450638
[ROCm] ci build fixes
2024-12-02 06:49:01 -06:00
Peter Hawkins
81991d87c8
JAX release 0.4.35
2024-10-22 15:00:23 -04:00
Peter Hawkins
e4f3f8f064
Use libtpu releases rather than libtpu-nightly for jax[tpu].
...
PiperOrigin-RevId: 688632409
2024-10-22 11:47:07 -07:00
jax authors
1c6b0a9193
Merge pull request #24465 from jakevdp:fix-mypy
...
PiperOrigin-RevId: 688632024
2024-10-22 11:45:27 -07:00
jax authors
9a2dd19a92
Merge pull request #21524 from andportnoy:aportnoy/unknown-platform-lowering-warning
...
PiperOrigin-RevId: 688630259
2024-10-22 11:40:39 -07:00
jax authors
1e41d5ef6f
Merge pull request #24452 from jakevdp:insert-doc
...
PiperOrigin-RevId: 688624762
2024-10-22 11:26:38 -07:00
