pytorch/xla

Pull Requests Commits

Skip non TPU devices

alanwaketan committed 1 year ago

48eef410

alanwaketan committed 1 year ago

fec38604

alanwaketan committed 1 year ago

a212189c

alanwaketan committed 1 year ago

aa68e604

alanwaketan committed 1 year ago

bccdacd9

add canonical index

alanwaketan committed 1 year ago

6f4f1709

fix the lowering

alanwaketan committed 1 year ago

38b28e70

Add a test case

alanwaketan committed 1 year ago

ad5b7a66

alanwaketan committed 1 year ago

12b5a7ed

Let grad ckpt apply opt-barrier to all params and buffers (#7206)

alanwaketan committed 1 year ago

Verified aec27305

Make clip_grad_norm_ follow the input's dtype (#7205)

alanwaketan committed 1 year ago

Verified f5c00624

openxla pin update to 06/05 (#7200)

zpcore committed 1 year ago

Verified 676853f2

[Pallas] Make _histogram output int32 (#7202)

alanwaketan committed 1 year ago

Verified 218f4e54

Update images to support dark mode (#7195)

loganthomas committed 1 year ago

Verified cdb4aec6

Implement XLASymNodeImpl::gt (#7192)

vanbasten23 committed 1 year ago

Verified 73346d7a

Re-land: `upsample_bilinear`: fix output data-type. (#7168)

ysiraichi committed 1 year ago

Verified e563cfef

Output tensorid before erroring out for empty data handle (#7176)

JackCaoG committed 1 year ago

Verified 376d6454

Warn when using persistent cache with debug env vars (#7175)

jonb377 committed 1 year ago

Verified bcf46c7a

[torch_xla2] Use `torch.Generator` for random number generation in `Environment` (#7018)

will-cromar committed 1 year ago

Verified 94fa9993

Update index for torch_xla apis (#7171)

JackCaoG committed 1 year ago

Verified 8102ad42

Run some of the examples in TPU CI (#7172)

JackCaoG committed 1 year ago

Verified 0d754931

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

Older