pytorch/xla

Pull Requests Commits

Add `tpu-info` as a dependency of `torch_xla[tpu]` (#7938)

will-cromar committed 1 year ago

Verified d69f3a54

Fix jaten for softmax(nan/inf). (#7944)

tqfan28 committed 1 year ago

Verified 9e3b4484

Op info test for expm1 .. fill (#7940)

qihqi committed 1 year ago

Verified 13affb9d

[torch_xla2] Ops fixes https://github.com/pytorch/xla/issues/7512 (#7936)

ShitongWang committed 1 year ago

Verified 87fe71ab

Double loader_prefetch_size and device_prefetch_size (#7934)

tengyifei committed 1 year ago

Verified a0b20edb

Support gmm and tgmm trace_pallas caching (#7921)

JackCaoG committed 1 year ago

Verified 8955571f

[benchmarks] Fix processing of PyTorch profiler events. (#7930)

ysiraichi committed 1 year ago

Verified 1b4b8281

[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

Newer Older