projects / platform / upstream / pytorch.git / commit
? search:
summary | shortlog | log | commit | commitdiff | tree
(parent: 1cd0252) | patch
Add BFloat16 support for cross, tril, triu, tril_indices, triu_indices and cumsum...
authorCaoE <e.cao@intel.com>
Tue, 14 Sep 2021 00:58:20 +0000 (17:58 -0700)
committerFacebook GitHub Bot <facebook-github-bot@users.noreply.github.com>
Tue, 14 Sep 2021 00:59:43 +0000 (17:59 -0700)
commit3855c246395303c2cde38c832a21cca577851b39
tree259d5df5e2cc63838cb191f4402bfa22b191c7cetree | snapshot
parent1cd0252eed8ddb26e4599ef2b0fec4d8843b8828commit | diff
Add BFloat16 support for cross, tril, triu, tril_indices, triu_indices and cumsum operators on CPU (#62454)

Summary:
Add BFloat16 support for cross, tril, triu, tril_indices, triu_indices and cumsum operators on CPU.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/62454

Reviewed By: albanD

Differential Revision: D30845805

Pulled By: heitorschueroff

fbshipit-source-id: f83836862e38109ec929e83567133e9e88096b8b
aten/src/ATen/native/TensorFactories.cpp diff | blob | history
aten/src/ATen/native/TriangularOps.cpp diff | blob | history
aten/src/ATen/native/cpu/CrossKernel.cpp diff | blob | history
test/test_tensor_creation_ops.py diff | blob | history
torch/testing/_internal/common_methods_invocations.py diff | blob | history
Domain: Machine Learning / ML Framework;