pytorch/xla

Pull Requests Commits

Update run_tests.sh skip pallas test on v4 due to lcoal failed too

ManfeiBai committed 1 year ago

3de6d81d

Update run_tests.sh for pallas tests

ManfeiBai committed 1 year ago

10b607da

Update run_tests.sh skip pallas test

ManfeiBai committed 1 year ago

518f6555

update libtpu version to 0821

ManfeiBai committed 1 year ago

9d4fbfa7

Update run_tests.sh

ManfeiBai committed 1 year ago

97d27510

Update run_tests.sh

ManfeiBai committed 1 year ago

db8bc935

ManfeiBai committed 1 year ago

82f32f6d

Update pjrt_computation_client_test.cc

ManfeiBai committed 1 year ago

edc8efe5

ManfeiBai committed 1 year ago

c3cddcc9

Update setup.py

ManfeiBai committed 1 year ago

fd990560

Update xla_util_test.cc

ManfeiBai committed 1 year ago

230e6921

Update pjrt_computation_client_test.cc

ManfeiBai committed 1 year ago

a3bf31e9

Update ifrt_computation_client_test.cc

ManfeiBai committed 1 year ago

6aca4809

Update @tsl//tsl/lib/core to @xla//xla/tsl/lib/core

ManfeiBai committed 1 year ago

bd18f51c

Update tsl path

ManfeiBai committed 1 year ago

338b0ea5

Update openxla-pin

ManfeiBai committed 1 year ago

5b4773b3

Implement bucketize in torch aten ops (#7396) (#8007)

simonteozw committed 1 year ago

Verified dbdebfd9

Set xla_tpu_enable_flash_attention=false to enable libtpu pin update (#8008)

bhavya01 committed 1 year ago

Verified 9c7f0835

[benchmarks] Disallow `XRT` for XLA:CUDA. (#8006)

ysiraichi committed 1 year ago

Verified afbd7bd5

Add nn.functional.adaptive_avg_pool2d (#8010)

tracych477 committed 1 year ago

Verified abb07e41

OpInfo test for take -- trace (#8009)

qihqi committed 1 year ago

Verified 183d1d41

add `histogram` .. `linalg.eigvals` (#8002)

ManfeiBai committed 1 year ago

Verified 24bc4b2a

Op info test for logdet, linalg.slogdet (#7503) (#7958) (#8004)

yenkwang committed 1 year ago

Verified 3a9f53f4

Follow 2 releases grace period for API deprecation (#8003)

zpcore committed 1 year ago

Verified 99924801

Use a different str for device instead of meta (#7995)

qihqi committed 1 year ago

Verified 1b57d1e8

Update test_xla_sharding.py to take into account the number of devices (#8001)

bhavya01 committed 1 year ago

Verified fc03aef3

Fixed Masked.normalize and masked.prod (#7999)

anishfish2 committed 1 year ago

Verified 0535c889

add `heaviside ` .. `bfloat16` (#7994)

ManfeiBai committed 1 year ago

Verified 2097a6bb

Add the scripts used for lightning docs (#7991)

qihqi committed 1 year ago

Verified e2488e7f

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

guyao committed 1 year ago

Verified 835a6a1f

Newer Older