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
Zhekun Zhang 8af3e50662
[Torch Dialect] Add support for AtenScalarTensorOp (#2085) 
* add scalar_tensor op

* add dynamo pass test; needs PR2062

* try to fix

* Empty commit, trigger test

* Empty commit, trigger test

* address comments

* use dtype function

* fix decompose rule

* remove unused include

* Empty commit, trigger test

* fix test

* disable ltc

* fix dtype

---------

Co-authored-by: zhekun.zhang <zhekun.zhang@bytedance.com>
 		2023-06-01 11:38:50 +08:00
..
CAPI [CAPI] add isValidSubtype to CAPI (#2127) 2023-05-13 22:15:45 -05:00
Conversion [MLIR][TORCH] Add support for the total_weight for aten.nll_loss_forward op 2023-05-30 20:29:27 +05:30
Dialect [Torch Dialect] Add support for AtenScalarTensorOp (#2085) 2023-06-01 11:38:50 +08:00
RefBackend Add complex dtype support on refbackend 2023-05-11 21:29:07 +05:30
CMakeLists.txt mhlo: migrate conversion to stablehlo (#1840) 2023-02-02 07:29:47 -06:00
InitAll.cpp [Stablehlo] add e2e test for aten.batch_norm (#2129) 2023-05-17 09:04:40 -07:00