pytorch/xla

Pull Requests Commits

Pin update (#7862)

qihqi committed 1 year ago

Verified 6e8e7084

Enable tests for index_copy, index_fill, and index_put. (#7903)

cornmander committed 1 year ago

Verified e4de7e54

Pass local_world_size into neuron.initialize_env (#7852)

jeffhataws committed 1 year ago

Verified 5867d395

Optimize _split_xla_args_tensor_sym_constant (#7900)

JackCaoG committed 1 year ago

Verified 20c33f9a

Introduce the `scan` interface (#7848)

tengyifei committed 1 year ago

Verified 95f833a3

fix resnet examples (#7899)

JackCaoG committed 1 year ago

Verified df1f994e

CPU time optimization for GraphInputMatcher (#7895)

JackCaoG committed 1 year ago

Verified ae28308b

update readme for installation (#7891)

zpcore committed 1 year ago

Verified 313c9084

Op info test for `special.i1 .. special.modified_bessel_i0` (#7874)

ayshiff committed 1 year ago

Verified b13fb655

[torch_xla2] Fix `squeeze` operator (#7893)

guyao committed 1 year ago

Verified 6c357c22

[Op info test] Implemented the following op: exp2 (#7411) (#7894)

yenkwang committed 1 year ago

Verified 97e69de8

Allows Llama 3 example to run on TPUs with multiple VMs (16+) (XLA2) (#7888)

zmelumian committed 1 year ago

Verified c0c564e6

[test] re-enable lowering context test (#7887)

tengyifei committed 1 year ago

Verified 854d16e4

Support dynamic activation quant for per-channel quantized matmul (#7867)

lsy323 committed 1 year ago

Verified 4bd2df1e

[Op info test] Implemented the following ops: select_scatter and signbit (#7892)

yenkwang committed 1 year ago

Verified 5755d77d

DYNAMO RNG seed update optimization (#7884)

JackCaoG committed 1 year ago

Verified 3776cccc

Reenable test_traceable_collectives.py (#7881)

JackCaoG committed 1 year ago

Verified d284a21e

Fix few issues found during copybara import to Google3. (#7880)

qihqi committed 1 year ago

Verified f8905a32

fix wheel name for pip==24.2 (#7883)

zpcore committed 1 year ago

Verified defe6151

[torch_xla2] Add support for torch.qr and torch.linalg.qr (#7885)

danielvegamyhre committed 1 year ago

Verified b162b66d

update wheel name (#7882)

zpcore committed 1 year ago

Verified e30927a5

[torch_xla2] Fix all aten jax ops getter (#7868)

chunnienc committed 1 year ago

Verified b075cf81

fix benchamrk batch_size report (#7877)

zpcore committed 1 year ago

Verified 940a4b1b

[torchxla2] enable exp to cfloat (#7861)

ManfeiBai committed 1 year ago

Verified bda425fa

Fix `norm` data-type when using AMP. (#7878)

ysiraichi committed 1 year ago

Verified ac13bf23

[torch_xla2] Add randint_like (#7879)

danielvegamyhre committed 1 year ago

Verified 51b8b70f

[torch_xla2] Support operator `ones_like`, `zeros_like`, `full_like` (#7875)

guyao committed 1 year ago

Verified bbe19926

[torch_xla2] Enable test_ops frac..hstack (#7871)

gmittert committed 1 year ago

Verified 51704c6a

update nightly wheel naming (#7873)

zpcore committed 1 year ago

Verified ac7fd44f

As_stride should aplly on input tensor, not the base tensor (#7864)

JackCaoG committed 1 year ago

Verified 0e35022f

Newer Older