pytorch/xla

Pull Requests Commits

Update test_while_loop.py

ManfeiBai committed 2 years ago

Verified cd728165

Update test_while_loop.py for test case

ManfeiBai committed 2 years ago

Verified 0a3af2ff

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

guyao committed 2 years ago

Verified 835a6a1f

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

will-cromar committed 2 years ago

Verified e6f1bde2

Add on_ready_event (#7984)

JackCaoG committed 2 years ago

Verified 8888217e

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

ManfeiBai committed 2 years ago

Verified 6ff03e6d

Fixed Masked norm op implementation (#7985)

anishfish2 committed 2 years ago

Verified 5c4bac58

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

simonteozw committed 2 years ago

Verified e3f6e9e4

[torch_xla2] Fix native_layer_norm lowering (#7981)

chunnienc committed 2 years ago

Verified 63733201

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

will-cromar committed 2 years ago

Verified a89f0a41

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

ManfeiBai committed 2 years ago

Verified a21924aa

Implement triangular_solve op. (#7980)

cornmander committed 2 years ago

Verified 33029ad8

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

ManfeiBai committed 2 years ago

Verified e9d537eb

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

zmelumian committed 2 years ago

Verified ef6b3703

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

guyao committed 2 years ago

Verified 12e59584

Update Neuron initializations (#7952)

jeffhataws committed 2 years ago

Verified 901c3a32

grid_sampler_3d, gaussian_nll_loss, dropout (#7969)

qihqi committed 2 years ago

Verified 900296aa

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

will-cromar committed 2 years ago

Verified 0528c9d7

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

will-cromar committed 2 years ago

Verified c201ec11

Revert "Allow MpDeviceLoader to shard dictionaries of tensors" (#7964)

JackCaoG committed 2 years ago

Verified 59570c77

Unpin `scipy` in upstream CI image (#7963)

will-cromar committed 2 years ago

Verified 282bf932

Run ansible release stage after local builds are installed (#7955)

will-cromar committed 2 years ago

Verified 8bebeddd

Remove numpy pin for upstream image (#7954)

will-cromar committed 2 years ago

Verified 92f8cfdf

Op info test for logaddexp, logaddexp2 and logcumsumexp (#7506) (#7957)

yenkwang committed 2 years ago

Verified 1ae32bf9

Update upstream docker image to Python 3.10 (#7953)

will-cromar committed 2 years ago

Verified 1a417b4b

Op info test for special.modified_bessel_i1 .. special.polygamma (#7928)

ayshiff committed 2 years ago

Verified e7653b85

Allow MpDeviceLoader to shard dictionaries of tensors (#7912)

bhavya01 committed 2 years ago

Verified 5acecc33

`DynamicShapeDetector` with trie implementation. (#7918)

ysiraichi committed 2 years ago

Verified 400bd91b

support dist.reduce_scatter_tensor (#7950)

zpcore committed 2 years ago

Verified b1fbb749

Skip dynamo collective tests on TPU v2 CI (#7949)

will-cromar committed 2 years ago

Verified f45775a5

Older