Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

Use Tuple instead of tuple in scan.py #8066

Closed
wants to merge 50 commits into from
Closed

Use Tuple instead of tuple in scan.py #8066

wants to merge 50 commits into from

Conversation

tengyifei
Copy link
Collaborator

No description provided.

ManfeiBai and others added 30 commits September 9, 2024 14:15
@ManfeiBai
Add Build Trigger for 2.5-rc1 release (#7978)
e9d537e
@cornmander
Implement triangular_solve op. (#7980)
33029ad
@ManfeiBai
Op Info tests broadcast_tensors, count_nonzero, cov, cross, a…
a21924a 
…nd `equal` (#7951)
@will-cromar
[torch_xla2] Fix type promotions by aten::copy_ and `aten::rand_lik…
a89f0a4 
…e` (#7960)
@chunnienc
[torch_xla2] Fix native_layer_norm lowering (#7981)
6373320
@simonteozw
Clarify instructions and fix small spelling error in README (#7975)
e3f6e9e 
Co-authored-by: Simon Teo <simonteo@google.com>
@anishfish2
Fixed Masked norm op implementation (#7985)
5c4bac5
@ManfeiBai
Update torchxla2 installation for torch on mac and others (#7990)
6ff03e6
@JackCaoG
Add on_ready_event (#7984)
8888217
@will-cromar
[torch_xla2] Use openwebtext for DDP example (#7959)
e6f1bde
@guyao
[torch_xla2] Op coverage - torch.diag (#7961)
835a6a1
@qihqi
Add the scripts used for lightning docs (#7991)
e2488e7
@ManfeiBai
add heaviside .. bfloat16 (#7994)
2097a6b
@anishfish2
Fixed Masked.normalize and masked.prod (#7999)
0535c88
@bhavya01
Update test_xla_sharding.py to take into account the number of devices (
fc03aef 
#8001)
@qihqi @will-cromar
Use a different str for device instead of meta (#7995)
1b57d1e 
Co-authored-by: Will Cromar <wcromar@google.com>
@zpcore
Follow 2 releases grace period for API deprecation (#8003)
9992480
@yenkwang
Op info test for logdet, linalg.slogdet (#7503) (#7958) (#8004)
3a9f53f
@ManfeiBai
add histogram .. linalg.eigvals (#8002)
24bc4b2
@qihqi
OpInfo test for take -- trace (#8009)
183d1d4
@tracych477
Add nn.functional.adaptive_avg_pool2d (#8010)
abb07e4
@ysiraichi
[benchmarks] Disallow XRT for XLA:CUDA. (#8006)
afbd7bd
@bhavya01
Set xla_tpu_enable_flash_attention=false to enable libtpu pin update (#…
9c7f083 
…8008)
@simonteozw
Implement bucketize in torch aten ops (#7396) (#8007)
dbdebfd 
Co-authored-by: Simon Teo <simonteo@google.com>
@ManfeiBai
Update Openxla-pin to Sep13, libtpu-pin to Sep13, jax to 0.4.33_night…
37894b0 
…ly (#7916)
@guyao
Implement operations cummax, cummin (#8013)
2a5897d
@nupurbaghel
Implement dist in torch aten ops (#8019)
1b1eea2
@matinehAkhlaghinia
Add fixes for narrow_copy and narrow ops (#8021)
008004d
@qihqi
[torchxla2] Op Info test for fft (#8018)
3f4b31b
@bhavya01
Update nightly version in README.md (#8024)
5ec5139
ManfeiBai and others added 20 commits September 16, 2024 12:32
@ManfeiBai
add masked.amin .. masked.mean (#8014)
0e47739
@guyao
Fix op test for `nn.functional.scaled_dot_product_attention"nn.functi… (
3b0097d 
#8025)
@ysiraichi
Remove contiguity assertion from XLATensorImpl. (#7998)
03374cd
@JackCaoG
Cache flash attention tracing (#8026)
8b93611
@nupurbaghel
Implement empty_like in torch aten ops (#8029)
0da723f
@cornmander
Fix unbind op. (#8033)
4fe55f5
@cornmander
Uncomment tests that are passing. (#8035)
0415cc6
@matinehAkhlaghinia
Fix quantile, nanquantile and gather (#8040)
6b33f8f
@simonteozw
Implement torch diagflat, and movedim as well as it is also needed as…
b79a46b 
… a helper function (#7378) (#8038)
@anishfish2
Fixed masked_scatter and masked_select (#8037)
5267354
@simonteozw
Fix minor grammar error (#8043)
c0501f0
@yenkwang
Op info test for linalg.norm, linalg.pinv, linalg.solve (#7503) (#8044)
441098b
@rpsilva-aws
Update and short circuit the Neuron initialization (#8041)
d0ea5cc
@dudulightricks
parallel_loader: Fix TPU memory leak when calling __iter__. (#8039)
d79a37c
@dvhg
[torch_xla2] Fix scatter and scatter_add (#8045)
2aea97a
@dvhg
[torch_xla2] Fix nn.functional.conv2d and conv3d (#8048)
6af550a
@will-cromar
[torch_xla2] Basic DDP example in a notebook (#8042)
ea8c47a
@jeffhataws
Update tests for Neuron (#8053)
e657c87
@ManfeiBai
add resize_, resize_as_, rot90, repeat_interleave, `combinati…
47d231f 
…ons`, `diag_embed` (#8036)
@tengyifei
Use Tuple instead of tuple in scan.py
0cd4db2
@ManfeiBai
Copy link
Collaborator

Thanks, could we merge to r2.5 branch to unblock upstream test? current PR is changing master branch

@tengyifei tengyifei changed the base branch from master to r2.5 September 25, 2024 00:39
@tengyifei tengyifei closed this Sep 25, 2024
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Reviewers

@ManfeiBai ManfeiBai Awaiting requested review from ManfeiBai

@JackCaoG JackCaoG Awaiting requested review from JackCaoG JackCaoG is a code owner

@will-cromar will-cromar Awaiting requested review from will-cromar will-cromar is a code owner

@lsy323 lsy323 Awaiting requested review from lsy323 lsy323 is a code owner

Assignees
No one assigned
Labels
None yet
Projects
None yet
Milestone
No milestone
Development

Successfully merging this pull request may close these issues.