pytorch/xla

Pull Requests Commits

JackCaoG committed 326 days ago

35111553

remove test_steps because we already have similar test in eager/test_eager_with_xla_compile.py

JackCaoG committed 326 days ago

91863c3d

add eager_mode_context back

JackCaoG committed 327 days ago

77a1450c

Refactor torch_xla.experimental.compile and move it out of experimental

JackCaoG committed 327 days ago

4d929148

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

will-cromar committed 327 days ago

Verified 2e1ebfc0

fix TPU test regarding word_size (#7743)

zpcore committed 327 days ago

Verified d77bf6fe

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

JackCaoG committed 327 days ago

Verified df59b1c3

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

JackCaoG committed 327 days ago

Verified 735dd435

Fix the SPMD doc strng (#7718)

JackCaoG committed 327 days ago

Verified b4d82588

Update readme with 2.4 release version (#7742)

bhavya01 committed 327 days ago

Verified 2c770683

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

ysiraichi committed 327 days ago

Verified 9d113adf

deprecate xrt_world_size (#7679)

zpcore committed 328 days ago

Verified 7f8ef791

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

will-cromar committed 328 days ago

Verified fb9f54ed

Optimize dynamo dynamic shape caching (#7726)

JackCaoG committed 328 days ago

Verified 33f69ad8

Do not try to trim the graph (#7725)

JackCaoG committed 328 days ago

Verified a34d28b4

Make CUDA OpenXLA fallback the default. (#7630)

ysiraichi committed 328 days ago

Verified 806de832

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

ysiraichi committed 328 days ago

Verified c69742c0

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

will-cromar committed 328 days ago

Verified c46cdf23

Add final 2.4 trigger (#7732)

bhavya01 committed 328 days ago

Verified 807c595d

[torch_xla2] Opinfo tests for `rsqrt`, `scalar_tensor`, `i0`, `slice*`, `scalar.window.*` (#7716)

will-cromar committed 328 days ago

Verified 2106ea0d

Build CUDA plugin before `torch` and `torch_xla` in nightly build (#7721)

will-cromar committed 329 days ago

Verified 42aa7bdc

Add support for dynamic shape in dynamo (#7676)

wonjoolee95 committed 330 days ago

Verified 2b6b4613

Update contributing doc after walkthrough (#7332)

tengyifei committed 331 days ago

Verified 5ea4535f

Unify the logics to check eager mode (#7709)

JackCaoG committed 333 days ago

Verified 4ba63ffd

Update OpenXLA pin as of 07172024 (#7715)

wonjoolee95 committed 333 days ago

Verified ea848830

Update eager.md (#7710)

JackCaoG committed 333 days ago

Verified 0ddc63a9

Add trapz, trapezoid, div (#7706)

ManfeiBai committed 333 days ago

Verified 2f024274

Add op support from `nan_to_num` to `movedim` (#7712)

miladm committed 333 days ago

Verified 35c537af

Add `tril`, `tile` ops support (#7711)

miladm committed 333 days ago

Verified a0310705

Change c10::optional to std::optional (#7702)

cyyever committed 334 days ago

Verified c5c49578

Older