pytorch/xla

Pull Requests Commits

Surface more general reduce API instead of CRS.

dlibenzi committed 6 years ago

141b6284

Switch to override leaf only (#1324)

ailzhang committed 6 years ago

Verified 3eee804d

Update MNIST and ResNet18 notebooks to MP API (#1347)

jysohn23 committed 6 years ago

Verified 0d95c38e

Add overrides for bitwise_xor_not. (#1232)

ailzhang committed 6 years ago

Verified 227111ab

Add XLA token based CRS deterministic sequencing.

dlibenzi committed 6 years ago

1d0ec090

Sets use_full_mat_mul_precision in Torch tests.

mruberry committed 6 years ago

Verified 2a41ea06

Set back the new XLA tensor when copy_() changed it.

dlibenzi committed 6 years ago

77966a6f

Group reduces by XLA type.

dlibenzi committed 6 years ago

4b9543d2

Allow multiple spawn in Colab.

dlibenzi committed 6 years ago

637f899f

Download data into different dir per proccess

jysohn23 committed 6 years ago

d6cb648f

move test_utils to torch_xla/test

zcain117 committed 6 years ago

ee8a4a30

Another pull of TF bits for the day!

dlibenzi committed 6 years ago

da7ffae0

Save metrics (#1291)

zcain117 committed 6 years ago

Verified c9b2a8ca

Update xmp.spawn() API to be able to start with fork (#1334)

jysohn23 committed 6 years ago

Verified 95f76eb9

Enable vardic AllReduce to allow XLA to adopt a deterministic scheduling across cores.

dlibenzi committed 6 years ago

f90f2333

Creates notice about handling shared weights (#1335)

mruberry committed 6 years ago

Verified 7708465a

Bring new TF with Tuple capable AllReduce.

dlibenzi committed 6 years ago

45b0adf6

Link PT BFloat16 to XLA BF16.

dlibenzi committed 6 years ago

d7b5521a

Actually return the vector from the function.

dlibenzi committed 6 years ago

ee980b48

Add sum_to_size to list. (#1329)

ailzhang committed 6 years ago

Verified 02184e77

Minor cleanup and comment.

dlibenzi committed 6 years ago

209713d2

Remove override for ones/zeros/full/empty_like/full_like/ones_l… (#1323)

ailzhang committed 6 years ago

Verified bb8b895a

Makes PyTorch Python tests opt-out instead of opt-in, extends testing (#1320)

mruberry committed 6 years ago

Verified 278f7c72

Remove override for eye and add override for eye_out. (#1317)

ailzhang committed 6 years ago

Verified a2867040

Remove override for randperm and add override for randperm_out.

Ailing Zhang committed 6 years ago

dbaf4bea

loop over self.env_vars which is a list (#1322)

taylanbil committed 6 years ago

Verified 842a8047

Prepare for the new vardic AllReduce.

dlibenzi committed 6 years ago

52758841

Remove override for arange. (#1315)

ailzhang committed 6 years ago

Verified 8eea339a

Redo serialization patch. (#1318)

ailzhang committed 6 years ago

Verified 3c1f929e

Remove override for bartlett_window/blackman_window/hamming_win… (#1314)

ailzhang committed 6 years ago

Verified 5a027bf1

Older