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
Sean Silva 8860b5c55d Add `torch.prim.If` 
This removes the use of `scf.if`, which required laundering back and
forth between `i1` and `!torch.bool` in the frontend. We will eventually
lower this op to `scf.if`, but this results in a cleaner IR and layering
at the frontend.
 		2021-06-16 14:04:31 -07:00
..
acap_dispatch.cpp Make C API files more consistent 2021-06-14 15:34:43 -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 Add `!torch.bool` type. 2021-06-16 13:22:00 -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 6d263b6f1c97fe6c45c75443e7daf6cd0c1c4222 2021-05-10 18:06:15 -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 Add `!torch.bool` type. 2021-06-16 13:22:00 -07: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 Add `torch.prim.If` 2021-06-16 14:04:31 -07: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 Add `!torch.bool` type. 2021-06-16 13:22:00 -07:00
torch_to_mlir_utils.h Add `!torch.bool` type. 2021-06-16 13:22:00 -07:00