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/Torch
History
Ashay Rane 9ec4712516
types: allow bf16 as result type for various tensor ops (#798) 
Prior to this patch, the result type for several tensor operations could
only be float32, float64, or null.  This patch adds bf16 to the list of
allowed result types.
 		2022-04-26 11:55:58 -07:00
..
IR Fix for 0-size dim inferred incorrectly. 2022-03-30 16:36:15 -07:00
Transforms types: allow bf16 as result type for various tensor ops (#798) 2022-04-26 11:55:58 -07:00
Utils Fixes https://github.com/llvm/torch-mlir/issues/751 where `torch.bool` is parsed as signless `i1`. (#752) 2022-04-13 12:28:27 -05:00
CMakeLists.txt E2e support for aten.softmax.int and aten.embedding 2021-10-18 17:57:45 -04:00