llvm
/
torch-mlir
mirror of https://github.com/llvm/torch-mlir
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity
torch-mlir/lib
History
Ramiro Leal-Cavazos e6675a50d3 Add support for dtype argument in reduction ops 
Many reduction ops take as an argument an optional output dtype that
can change the type of the input tensor before the reduction is
performed. This commit adds support for the optional dtype flag that
had been previously ignored.

Test:
/tools/torchscript_e2e_test.sh -f 'ReduceSumDtype'
/tools/torchscript_e2e_test.sh -f 'ReduceSumDImIntListDtype'
 		2021-11-30 12:53:59 -05:00
..
CAPI Enable python extensions when building out of tree (#363) 2021-10-27 17:04:12 -07:00
Conversion Add support for dtype argument in reduction ops 2021-11-30 12:53:59 -05:00
Dialect Add support for dtype argument in reduction ops 2021-11-30 12:53:59 -05:00
RefBackend Add support for multiple return values 2021-11-16 21:07:45 -05:00
CMakeLists.txt Move external/torch-mlir to the root of the repo. 2021-09-27 17:11:08 -07:00
InitAll.cpp Dual license the torch-mlir project. 2021-10-01 10:46:08 -07:00