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/Dialect
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
..
Torch Add support for dtype argument in reduction ops 2021-11-30 12:53:59 -05:00
TorchConversion Update llvm upstream to 0b17336f793108a7b10c3fa913039144ef1d0f61 2021-11-16 13:04:51 -05:00
CMakeLists.txt Move external/torch-mlir to the root of the repo. 2021-09-27 17:11:08 -07:00