pytorch/xla

Pull Requests Commits

Update WORKSPACE

ManfeiBai committed 1 year ago

Verified d18bba2d

[benchmarks] Add IR dump option. (#7927)

ysiraichi committed 1 year ago

Verified beeb40b1

[Op info test] Implemented the following ops: log_softmax, softmax, s… (#7922)

yenkwang committed 1 year ago

Verified 9dfab228

[torch_xla2] Add oss only tags to jax env initialization (#7920)

chunnienc committed 1 year ago

Verified eedf2b9b

Update the guide for multi process persistent cache (#7917)

JackCaoG committed 1 year ago

Verified 17a4ef58

In dynamo optim_mode avoid unnecessary set_attr (#7915)

JackCaoG committed 1 year ago

Verified c58680cb

Add few sample docs (#7907)

qihqi committed 1 year ago

Verified 00e40632

Support `dist.all_gather` related collective ops (#7860)

zpcore committed 1 year ago

Verified f9a706e0

add dynamo config skip_input_data_check (#7913)

JackCaoG committed 1 year ago

Verified 98a93726

Update GPU nightly wheel link (#7910)

zpcore committed 1 year ago

Verified c400d93c

Move the dynamo bridge to the _dynamo repo (#7909)

JackCaoG committed 1 year ago

Verified a7533b0d

[OpenXLA-pin-update trigger] Create _openxla_pin_update_weekly.yml (#7108)

ManfeiBai committed 1 year ago

Verified 5f82da90

[torch_xla2] Fix max_pool2d_with_indices lowering (#7905)

chunnienc committed 1 year ago

Verified 64a1b7fc

Implement index_select op. (#7908)

cornmander committed 1 year ago

Verified 7cbedca3

[torch_xla2] Relax python version constraint to support py3.9 (#7906)

chunnienc committed 1 year ago

Verified 71906013

Fix the crash with copy op in dynamo (#7902)

JackCaoG committed 1 year ago

Verified 3d860bf4

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

Older