llvm
/
torch-mlir
mirror of https://github.com/llvm/torch-mlir
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity
torch-mlir/frontends/pytorch/csrc/builder
History
Yi Zhang bfc3ee35c6 Import Machine Translation model to MLIR. 
This includes the following changes to import MT model into MLIR. There
are still a lot of work to for actual compilation.
- Add `torch.dict<>`, `torch.any`, `torch.number` types
- Add `torch.prim.DictConstruct` op
- Fix `torch.prim.TupleConstruct` op assembly format to include resulting types
 		2021-08-10 15:22:06 -04:00
..
acap_dispatch.cpp Rework the python build to a static assembly of MLIR+NPCOMP (#251) 2021-07-27 16:10:10 -07:00
acap_dispatch.h Significantly restructure torch/aten import design. 2021-05-19 13:37:39 -07:00
class_annotator.cpp Introduce `!torch.tensor` / `!torch.vtensor` types. 2021-06-10 10:56:48 -07:00
class_annotator.h Introduce `!torch.tensor` / `!torch.vtensor` types. 2021-06-10 10:56:48 -07:00
debug.cpp Delete old PyTorch 1.3 type dispatch oriented code paths. 2020-11-12 22:27:05 -08:00
debug.h Delete old PyTorch 1.3 type dispatch oriented code paths. 2020-11-12 22:27:05 -08:00
func_builder.cpp Bump llvm-project to 5b2e7f50a6798fd9b9c79d9d62fdebcd9e78525b. (#260) 2021-07-29 12:26:54 -07:00
func_builder.h Add `torch.constant.int` and `torch.constant.float`. 2021-06-15 15:29:42 -07:00
function_importer.cpp Bump llvm-project to 5b2e7f50a6798fd9b9c79d9d62fdebcd9e78525b. (#260) 2021-07-29 12:26:54 -07:00
function_importer.h Basic infra for annotate shapes and dtypes on arguments. 2021-04-01 18:40:03 -07:00
ivalue_importer.cpp Import Machine Translation model to MLIR. 2021-08-10 15:22:06 -04:00
ivalue_importer.h Add ability to annotate TorchScript classes. 2021-02-25 11:28:34 -08:00
mlir_utils.h Significantly restructure torch/aten import design. 2021-05-19 13:37:39 -07:00
module_builder.cpp Improve diagnostic handler 2021-05-20 11:28:20 -07:00
module_builder.h Add ability to annotate TorchScript classes. 2021-02-25 11:28:34 -08:00
node_importer.cpp Import Machine Translation model to MLIR. 2021-08-10 15:22:06 -04:00
node_importer.h Properly model "derefinement". 2021-03-03 15:09:44 -08:00
op_builder.cpp Add `!torch.bool` type. 2021-06-16 13:22:00 -07:00
op_builder.h Add !torch.str type. 2021-06-15 10:10:59 -07:00
python_bindings.cpp Add recognition/folder/lowering for aten::__is__, aten::ne.int, and aten::dim 2021-04-30 10:57:02 -07:00
torch_to_mlir_utils.cpp Import Machine Translation model to MLIR. 2021-08-10 15:22:06 -04:00
torch_to_mlir_utils.h Add `!torch.int` type. 2021-06-17 07:28:23 -07:00