pytorch/xla

Pull Requests Commits

disable dynamo dynamic shape test since it is broken by some upstream change

JackCaoG committed 1 year ago

f9228eb5

openxla pin update (#7758)

zpcore committed 1 year ago

Verified dc3b265a

clean up deprecated (#7753)

zpcore committed 1 year ago

Verified 5702dae1

Refactor torch_xla.experimental.compile and move it out of experimental (#7750)

JackCaoG committed 1 year ago

Verified bdd00e54

Make libtpu install more explict (#7737)

JackCaoG committed 1 year ago

Verified 009e31aa

Fix few ops (#7759)

qihqi committed 1 year ago

Verified 2b5538d6

Add a guide to set up Visual Studio Code with PyTorch/XLA (#7755)

will-cromar committed 1 year ago

Verified b2e5c7d2

Move fsdp autowrap test to v4 TPUCI only (#7751)

JackCaoG committed 1 year ago

Verified db7fada8

Fix tril (#7744)

qihqi committed 1 year ago

Verified f7413b43

Remove `torch_xla[tpuvm]` optional dep (#7736)

will-cromar committed 1 year ago

Verified 44d637ad

[torch_xla2] Implement `bessel_*`, `chebyshev_*`, `hermite_*`, `erfcx`, `i0e` (#7723)

will-cromar committed 1 year ago

Verified 07a83636

add test for dynamo + traceable collectives (#7745)

JackCaoG committed 1 year ago

Verified b9c19b45

[benchmarks] Turn CUDAGraphs on for inductor. (#7749)

ysiraichi committed 1 year ago

Verified 0b990857

[benchmarks] Set matrix multiplication precision. (#7748)

ysiraichi committed 1 year ago

Verified 2870e93b

[benchmarks] Fix batch size logic. (#7747)

ysiraichi committed 1 year ago

Verified 62807012

Support `@xla.step` decorator without parentheses (#7746)

will-cromar committed 1 year ago

Verified 2e1ebfc0

fix TPU test regarding word_size (#7743)

zpcore committed 1 year ago

Verified d77bf6fe

Fix the crash when symint is part of the output with dynamic torch.compile (#7735)

JackCaoG committed 1 year ago

Verified df59b1c3

Use torch_xla.experimental.compile for all examples (#7642)

JackCaoG committed 1 year ago

Verified 735dd435

Fix the SPMD doc strng (#7718)

JackCaoG committed 1 year ago

Verified b4d82588

Update readme with 2.4 release version (#7742)

bhavya01 committed 1 year ago

Verified 2c770683

Rename `aten_cpu_fallback` into `aten_fallback`. (#7646)

ysiraichi committed 1 year ago

Verified 9d113adf

deprecate xrt_world_size (#7679)

zpcore committed 1 year ago

Verified 7f8ef791

Extend timeout for `TpuInfoCliTest.test_single_process_e2e_one_chip` (#7733)

will-cromar committed 1 year ago

Verified fb9f54ed

Optimize dynamo dynamic shape caching (#7726)

JackCaoG committed 1 year ago

Verified 33f69ad8

Do not try to trim the graph (#7725)

JackCaoG committed 1 year ago

Verified a34d28b4

Make CUDA OpenXLA fallback the default. (#7630)

ysiraichi committed 1 year ago

Verified 806de832

Re-implement and refactor the verifier. (#7724)

ysiraichi committed 1 year ago

Verified c69742c0

[torch_xla2] Add decorator to promote int inputs (#7729)

will-cromar committed 1 year ago

Verified c46cdf23

Add final 2.4 trigger (#7732)

bhavya01 committed 1 year ago

Verified 807c595d

Older