pytorch/xla

Pull Requests Commits

Fix TestXLATensor.test_log_softmax on TPU

asuhan committed 6 years ago

acb7cb7e

[XLATensor] Make convolution attributes more general

asuhan committed 6 years ago

c1696855

[XLATensor] Clean up XLATensor::size

asuhan committed 6 years ago

75fb4360

Fix gratuitous use of torch.Tensor in tests

asuhan committed 6 years ago

fd032d06

Some command line args for run_tests.sh

dlibenzi committed 6 years ago

69e157a4

Rename TestXLATensor.log_softmax to TestXLATensor.test_log_softmax

asuhan committed 6 years ago

f7fc4cca

Make flags clang dependent.

dlibenzi committed 6 years ago

30f1dc7b

Update the PyTorch commit id to latest

asuhan committed 6 years ago

07ff8dd7

Make addmm parameters const refs

asuhan committed 6 years ago

6fced970

Move big tensor method implementations to helpers

asuhan committed 6 years ago

4c97a0a7

[XLATensor] Add addmm

asuhan committed 6 years ago

6c8298dc

[XLATensor] Add avg_pool2d

asuhan committed 6 years ago

d17b883a

[XLATensor] Add size method

asuhan committed 6 years ago

f501d7eb

Expand tensor type mapping to the full list of supported tensor types.

dlibenzi committed 6 years ago

45f42939

Some more code reuse.

dlibenzi committed 6 years ago

ae0e00d4

dlibenzi committed 6 years ago

9adbcecd

Some more code refactoring.

dlibenzi committed 6 years ago

c63ed9b7

More refactoring.

dlibenzi committed 6 years ago

83f6611f

Further optimizations and cleanups.

dlibenzi committed 6 years ago

a89df65a

More refinements to the tensor interface.

dlibenzi committed 6 years ago

7969b4f8

Added more fine grained support for PT<->XLA integer types.

dlibenzi committed 6 years ago

88c908a7

Further tensor util code simplification.

dlibenzi committed 6 years ago

b823e80d

Be consistent in the IR Node ToString value separation.

dlibenzi committed 6 years ago

1c6c3bb8

Added C++ test utility to test on all available device types.

dlibenzi committed 6 years ago

73a3f36c

Added C++ unittest util file and added namespaces.

dlibenzi committed 6 years ago

728540bd

Redirect C++ test log to file instead of suppressing it.

dlibenzi committed 6 years ago

84266bb4

Made C++ tensor test work and removed logging when testing.

dlibenzi committed 6 years ago

e07d5d4d

Added simple tensor test (missing check as ATEN throws) and minor refactoring of XLA tensor.

dlibenzi committed 6 years ago

86d684c8

Made literal to tensor conversion use CopyTensors, which is case or relayout is considerably faster.

dlibenzi committed 6 years ago

9668dd45

dlibenzi committed 6 years ago

70ef4e84

Newer Older