pytorch/xla

Pull Requests Commits

Update index for torch_xla apis

JackCaoG committed 1 year ago

7cdd259e

isolate the PJRT dependency between torch_xla and torch_xla2 (#7144)

zpcore committed 1 year ago

Verified 0f9bff34

`mul`: convert inputs to result type. (#7130)

ysiraichi committed 1 year ago

Verified 7938bb5d

Unify how we use torch commit in the CI. (#7164)

vanbasten23 committed 1 year ago

Verified 89832c3d

Reuse DLDeviceType. (#7163)

vanbasten23 committed 1 year ago

Verified 09cfdbf4

Add a CI workflow for tests that requires pytorch CUDA. (#7140)

vanbasten23 committed 1 year ago

Verified 6fadbf5d

Update configuration.yaml (#7158)

JackCaoG committed 1 year ago

Verified 8471826a

[Doc] Update spmd.md for doc (#7019)

ManfeiBai committed 1 year ago

Verified cb482bca

Add optimizer priming for dist chkpt (#6572)

jonb377 committed 1 year ago

Verified 8fd051f2

add PT_XLA_DEBUG_LEVEL (#7149)

JackCaoG committed 1 year ago

Verified 8c2234ee

Make from_dlpack handle cuda synchronization implicitly for input tensors that have __dlpack__ and __dlpack_device__ attributes. (#7125)

vanbasten23 committed 1 year ago

Verified daada224

[Pallas] Introduce GMM(torch.autograd.Function) (#7152)

alanwaketan committed 1 year ago

Verified aeed61a9

Deprecate XLA_USE_BF16 (#7150)

JackCaoG committed 1 year ago

Verified af51f063

[Pallas] Introduce gmm_backward (#7151)

alanwaketan committed 1 year ago

Verified c96c95a4

[Pallas] Make repeat_with_fixed_output_size not OOM on VMEM (#7145)

alanwaketan committed 1 year ago

Verified ce1205e1

Remove old checkout files from GitHub workspace (#7146)

bhavya01 committed 1 year ago

Verified ffbbd438

Disable kokoro and Update without key (#7106)

ManfeiBai committed 1 year ago

Verified 0ca090c3

add test for onehot to make sure no fallback (#7127)

JackCaoG committed 1 year ago

Verified ad73e070

Add Python 3.11 version for release build (#7143)

wonjoo-wj committed 1 year ago

Verified bd70d3f0

Revert "`upsample_bilinear`: fix output data-type." (#7142)

JackCaoG committed 1 year ago

Verified f1141980

[Pallas] Support tgmm (#7137)

alanwaketan committed 1 year ago

Verified 50d81b3e

Update contribute to include bdist_wheel (#7135)

JackCaoG committed 1 year ago

Verified c367b66f

Facebook->Meta in README (#7141)

kit1980 committed 1 year ago

Verified 0eded8da

Add function for retrieving fallback operations. (#7116)

ysiraichi committed 1 year ago

Verified 6d271232

`upsample_bilinear`: fix output data-type. (#7111)

ysiraichi committed 1 year ago

Verified 468a5c93

update example dir's README (#7136)

JackCaoG committed 1 year ago

Verified c7bbdfbb

Update XLA pin as of 20240528 (#7131)

wonjoo-wj committed 1 year ago

Verified 6f406b77

[FSDPv2] Shard on the maximal dim of weights (#7134)

alanwaketan committed 1 year ago

Verified 15fc0f1c

[Pallas] Make gmm support bf16 (#7133)

alanwaketan committed 1 year ago

Verified fb373129

[MoE] Test sorting lhs for gmm (#7121)

alanwaketan committed 1 year ago

Verified 8531d1c5

Older