Skip to content

Use from_dlpack tweak from @JesseFarebro#3

Open
lebrice wants to merge 8 commits into
masterfrom
torch_to_jax_tweak
Open

Use from_dlpack tweak from @JesseFarebro#3
lebrice wants to merge 8 commits into
masterfrom
torch_to_jax_tweak

Try to work arround AssertionError in XLA

cb3536b
Select commit
Loading
Failed to load commit list.
Codecov / codecov/patch failed May 1, 2025 in 0s

43.75% of diff hit (target 89.15%)

View this Pull Request on Codecov

43.75% of diff hit (target 89.15%)

Annotations

Check warning on line 146 in torch_jax_interop/to_jax.py

See this annotation in the file changed.

@codecov codecov / codecov/patch

torch_jax_interop/to_jax.py#L146

Added line #L146 was not covered by tests

Check warning on line 151 in torch_jax_interop/to_jax.py

See this annotation in the file changed.

@codecov codecov / codecov/patch

torch_jax_interop/to_jax.py#L149-L151

Added lines #L149 - L151 were not covered by tests

Check warning on line 158 in torch_jax_interop/to_jax.py

See this annotation in the file changed.

@codecov codecov / codecov/patch

torch_jax_interop/to_jax.py#L154-L158

Added lines #L154 - L158 were not covered by tests