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/Conversion
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
..
TorchToLinalg Add support for dtype argument in reduction ops 2021-11-30 12:53:59 -05:00
TorchToSCF Update llvm upstream to 0b17336f793108a7b10c3fa913039144ef1d0f61 2021-11-16 13:04:51 -05:00
TorchToStd add aten.add.int lowering in TorchToStd 2021-11-29 13:22:50 -05:00
TorchToTosa Update llvm upstream to 0b17336f793108a7b10c3fa913039144ef1d0f61 2021-11-16 13:04:51 -05:00
CMakeLists.txt Add a basic TOSA E2E backend. 2021-10-08 09:59:45 -07:00
PassDetail.h Dual license the torch-mlir project. 2021-10-01 10:46:08 -07:00
Passes.cpp Add a basic TOSA E2E backend. 2021-10-08 09:59:45 -07:00