pytorch/xla

Pull Requests Commits

Update test_fori_loop.py

ManfeiBai committed 1 year ago

Verified 853a6f26

Update test_fori_loop.py

ManfeiBai committed 1 year ago

Verified 7e1ffcbc

Update test_fori_loop.py

ManfeiBai committed 1 year ago

Verified cea90628

Create test_fori_loop.py

ManfeiBai committed 1 year ago

Verified a1ed583d

Update fori_loop.py

ManfeiBai committed 1 year ago

Verified dcf65fc4

Update fori_loop.py

ManfeiBai committed 1 year ago

Verified f0dd53e6

Update init_python_bindings.cpp

ManfeiBai committed 1 year ago

Verified c62ad7d4

Update init_python_bindings.cpp

ManfeiBai committed 1 year ago

Verified 0a0f8c66

Update init_python_bindings.cpp

ManfeiBai committed 1 year ago

Verified 8ac0bb5b

Update fori_loop.py

ManfeiBai committed 1 year ago

Verified c4b43844

Create fori_loop.py

ManfeiBai committed 1 year ago

Verified 0a2cc00c

Add CUDA as supported xla backends to github issue template (#6517)

vanbasten23 committed 1 year ago

Verified e736de87

Misc fix for jax export, ran formatter (#6516)

qihqi committed 1 year ago

Verified 1bcc0145

Lower the embedding op (#6495)

bhavya01 committed 1 year ago

Verified 657b6925

Lowering for _pdist_forward (#6507)

zpcore committed 1 year ago

Verified a5856f95

Enable tests for reflection_pad1d, reflection_pad3d, grid_sampler_2d (#6513)

qihqi committed 1 year ago

Verified 2797852a

Fix some more core aten ops (#6519)

wonjoo-wj committed 1 year ago

Verified b198a48b

[benchmarks] Fix execution with AMP precision. (#6512)

ysiraichi committed 1 year ago

Verified 408b3766

[SPMD] input sharding for both train & test data loading in imagenet example (#6515)

yeounoh committed 1 year ago

Verified a5692c20

Read from constants if it exists (#6510)

qihqi committed 1 year ago

Verified 0fa24a13

[benchmarks] Increase compilation cache. (#6509)

ysiraichi committed 1 year ago

Verified 7a1ee8f8

add unbounded dynamism test for some aten ops, support dynamism on add (#6443)

lsy323 committed 1 year ago

Verified 8d91ff58

Correctly set the exit code when an exception is raised in the atexit callback _prepare_to_exit. (#6383)

vanbasten23 committed 1 year ago

Verified 1cd3f460

Reapply lower randperm (#6482)

changm committed 1 year ago

Verified 157e06e4

Fix a bug in convolution for torch_xla2. Enable tests (#6496)

qihqi committed 1 year ago

Verified b9352685

[benchmarks] Remove `--strict-compatible` argument. (#6491)

ysiraichi committed 1 year ago

Verified 6f4409fa

[FSDPv2] Use the global mesh API (#6500)

alanwaketan committed 1 year ago

Verified ef8f6c3f

[FSDPv2] Enable auto-wrapping (#6499)

alanwaketan committed 1 year ago

Verified cbe0ca84

[SPMD] Introduce global mesh (#6498)

alanwaketan committed 1 year ago

Verified 9b123349

CI: increase build timeout from 180 to 240 minutes (#6492)

cota committed 1 year ago

Verified 6ae122d3

Older