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
Abhishek Varma bb259f918a [MLIR][TORCH] Add lowering for `aten._softmax` when `half_to_float=True` 
-- This commit adds decompose logic for `aten._softmax` when
   `half_to_float` is `True`.
-- An e2e test case will be added once support for half to float conversion for
   `aten._softmax` is added upstream.

Signed-off-by: Abhishek Varma <abhishek@nod-labs.com>
 		2022-11-28 22:32:00 +05:30
..
IR Revert "build: update llvm tag to 147fe9de" 2022-11-25 12:41:56 +05:30
Transforms [MLIR][TORCH] Add lowering for `aten._softmax` when `half_to_float=True` 2022-11-28 22:32:00 +05:30
Utils mlir: add missing dependency on TableGen targets (#1537) 2022-11-01 14:59:11 -05:00
CMakeLists.txt E2e support for aten.softmax.int and aten.embedding 2021-10-18 17:57:45 -04:00