pytorch/xla

Pull Requests Commits

Update run_tests.sh

ManfeiBai committed 1 year ago

Verified 871bc4a1

Set xla_tpu_enable_flash_attention=false to enable libtpu pin update (#8008)

bhavya01 committed 1 year ago

Verified 9c7f0835

[benchmarks] Disallow `XRT` for XLA:CUDA. (#8006)

ysiraichi committed 1 year ago

Verified afbd7bd5

Add nn.functional.adaptive_avg_pool2d (#8010)

tracych477 committed 1 year ago

Verified abb07e41

OpInfo test for take -- trace (#8009)

qihqi committed 1 year ago

Verified 183d1d41

add `histogram` .. `linalg.eigvals` (#8002)

ManfeiBai committed 1 year ago

Verified 24bc4b2a

Op info test for logdet, linalg.slogdet (#7503) (#7958) (#8004)

yenkwang committed 1 year ago

Verified 3a9f53f4

Follow 2 releases grace period for API deprecation (#8003)

zpcore committed 1 year ago

Verified 99924801

Use a different str for device instead of meta (#7995)

qihqi committed 1 year ago

Verified 1b57d1e8

Update test_xla_sharding.py to take into account the number of devices (#8001)

bhavya01 committed 1 year ago

Verified fc03aef3

Fixed Masked.normalize and masked.prod (#7999)

anishfish2 committed 1 year ago

Verified 0535c889

add `heaviside ` .. `bfloat16` (#7994)

ManfeiBai committed 1 year ago

Verified 2097a6bb

Add the scripts used for lightning docs (#7991)

qihqi committed 1 year ago

Verified e2488e7f

[torch_xla2] Op coverage - torch.diag (#7961)

guyao committed 1 year ago

Verified 835a6a1f

[torch_xla2] Use openwebtext for DDP example (#7959)

will-cromar committed 1 year ago

Verified e6f1bde2

Add on_ready_event (#7984)

JackCaoG committed 1 year ago

Verified 8888217e

Update torchxla2 installation for torch on mac and others (#7990)

ManfeiBai committed 1 year ago

Verified 6ff03e6d

Fixed Masked norm op implementation (#7985)

anishfish2 committed 1 year ago

Verified 5c4bac58

Clarify instructions and fix small spelling error in README (#7975)

simonteozw committed 1 year ago

Verified e3f6e9e4

[torch_xla2] Fix native_layer_norm lowering (#7981)

chunnienc committed 1 year ago

Verified 63733201

[torch_xla2] Fix type promotions by `aten::copy_` and `aten::rand_like` (#7960)

will-cromar committed 1 year ago

Verified a89f0a41

Op Info tests `broadcast_tensors`, `count_nonzero`, `cov`, `cross`, and `equal` (#7951)

ManfeiBai committed 1 year ago

Verified a21924aa

Implement triangular_solve op. (#7980)

cornmander committed 1 year ago

Verified 33029ad8

Add Build Trigger for 2.5-rc1 release (#7978)

ManfeiBai committed 1 year ago

Verified e9d537eb

[torch_ xla2] LLama3 example - fixed sharding configuration on LLama3 example (#7973)

zmelumian committed 1 year ago

Verified ef6b3703

[torch_xla2] Fix the test of `cumsum` (#7965)

guyao committed 1 year ago

Verified 12e59584

Update Neuron initializations (#7952)

jeffhataws committed 1 year ago

Verified 901c3a32

grid_sampler_3d, gaussian_nll_loss, dropout (#7969)

qihqi committed 1 year ago

Verified 900296aa

Remove `TPU_LIBRARY_PATH` from legacy TPU CI (#7968)

will-cromar committed 1 year ago

Verified 0528c9d7

Increment version in upstream CI `docker push` command (#7967)

will-cromar committed 1 year ago

Verified c201ec11

Older