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
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
..
CAPI Fix Base Lazy Backend Type Conversion (#1412) 2022-10-04 15:53:28 -07:00
Conversion Revert "build: update llvm tag to 147fe9de" 2022-11-25 12:41:56 +05:30
Dialect [MLIR][TORCH] Add lowering for `aten._softmax` when `half_to_float=True` 2022-11-28 22:32:00 +05:30
RefBackend [cleanup] Make diagnostics better 2022-11-17 02:09:54 -08:00
CMakeLists.txt Resolve CI testing failure for Lazy Tensor Core (#1088) 2022-07-30 09:40:02 -04:00
InitAll.cpp Resolve CI testing failure for Lazy Tensor Core (#1088) 2022-07-30 09:40:02 -04:00