pytorch/xla

Pull Requests Commits

Update xla_util_test.cc

ManfeiBai committed 1 year ago

Verified ceed7dda

Update pjrt_computation_client_test.cc

ManfeiBai committed 1 year ago

Verified 9ec6abfb

Update ifrt_computation_client_test.cc

ManfeiBai committed 1 year ago

Verified f77f444e

ManfeiBai committed 1 year ago

Verified f346c1e2

Update run_tests.sh

ManfeiBai committed 1 year ago

Verified a7239f1c

Update setup.py

ManfeiBai committed 1 year ago

Verified 977154fb

Update WORKSPACE

ManfeiBai committed 1 year ago

Verified 6ed4ce41

fix dynamo inplace copy (#7933)

zpcore committed 1 year ago

Verified 989ac69c

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

Older