pytorch/xla

Pull Requests Commits

Adds XLA to device type testing framework (#1055)

mruberry committed 6 years ago

Verified 8d77c36e

enable mkldnn (#1053)

ailzhang committed 6 years ago

Verified 88c6eb99

IMplement xla_model side "first ordinal" discovery.

dlibenzi committed 6 years ago

6ce9affa

Mp imagenet scheduler (#1049)

zcain117 committed 6 years ago

Verified df6d66be

Imagenet dataset length (#1048)

zcain117 committed 6 years ago

Verified 8aa07906

Revise the metrics API and add more Python side accessibility.

dlibenzi committed 6 years ago

ce3bb9f9

add option for torchvision resnet18 (#1047)

zcain117 committed 6 years ago

Verified 4d7d9f45

demonstration of scheduler-style learning rate (#1012)

zcain117 committed 6 years ago

Verified bc14eb27

Actually print the tensor ID not the alias ID.

dlibenzi committed 6 years ago

ca854a83

Fix benchmark script.

dlibenzi committed 6 years ago

cdd7e63f

Minor adjustments to CONTRIBUTING.md

asuhan committed 6 years ago

79dca0ab

Fix typos in API_GUIDE.md

asuhan committed 6 years ago

a59c6f2e

[Ready]try add linter in circleci (#1041)

ailzhang committed 6 years ago

Verified b8943c0f

Add function to detect tensor potential aliases.

dlibenzi committed 6 years ago

4c76266d

Added more meaningful device spec print for multi-processing.

dlibenzi committed 6 years ago

68f9db8b

Added MP API documentation.

dlibenzi committed 6 years ago

45b9e4a2

add issue templates (#1036)

ailzhang committed 6 years ago

Verified 7d0abbc7

Use static for auto-unregistration object.

dlibenzi committed 6 years ago

429c89ad

draft contribution guide for discussion (#1033)

ailzhang committed 6 years ago

Verified fb6b0788

Fixed the XLA _out() code generation and added a few more overrides.

dlibenzi committed 6 years ago

040166bc

Ailing Zhang committed 6 years ago

a7e47149

Ailing Zhang committed 6 years ago

e672f3ec

Allow reduce operations in every point of the graph.

dlibenzi committed 6 years ago

acff0b9a

allow update any env (#1028)

taylanbil committed 6 years ago

Verified 8bc314b2

Added XLA multi-processing support and API.

dlibenzi committed 6 years ago

1f45f0a3

Make add_scatter_combiner a std::function

asuhan committed 6 years ago

271c6fcf

allow copy.copy() to return xla tensor (#1018)

ailzhang committed 6 years ago

Verified 2e67d89a

Fix conv backward hash population and ToString formatting.

dlibenzi committed 6 years ago

2ba4ec4c

Fix convolution ToString() formatting.

dlibenzi committed 6 years ago

856b864a

Override is_floating_point() and is_signed().

dlibenzi committed 6 years ago

da89b09c

Newer Older