pytorch/xla

Pull Requests Commits

Update test_core_aten_ops.py

ManfeiBai committed 1 year ago

Verified d7319fb2

Add limit to max inflight TPU computations (#6533)

will-cromar committed 1 year ago

Verified 29b01c6a

OpenXLA pin update (#6530)

yeounoh committed 1 year ago

Verified 27d1f703

Update TPU CI workflow to trigger on push to master and re-enable spm… (#6531)

mbzomowski committed 1 year ago

Verified 1d4a8da5

[FSDPv2] Move the module to xla device (#6525)

alanwaketan committed 1 year ago

Verified afc1f0e2

Modify export to use refactored _graph_module_flat_inputs (#6526)

qihqi committed 1 year ago

Verified 959e4786

aggregate.py: fix detection of no datapoints (#6523)

cota committed 1 year ago

Verified d6bc4f31

aggregate.py: fix relative error computation (#6522)

cota committed 1 year ago

Verified 7da7ee42

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

[export] Remove torch._export.export uses (#6486)

angelayi committed 1 year ago

Verified fba464b1

Remove random tests in test_core_aten_ops (#6436)

qihqi committed 1 year ago

Verified acf12a38

[benchmarks] Add `no-skip` argument. (#6484)

ysiraichi committed 1 year ago

Verified e8a5f007

Older