llvm
/
torch-mlir
mirror of https://github.com/llvm/torch-mlir
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity
torch-mlir/build_tools
History
Prashant Kumar 10c8e3c593 Add simple neural_net and bert_training scripts. 
1. With the help of `make_fx` we are able to get the full training graph
   with weight updates.
2. NeuralNet_training passes. Bert_training passes after cherry-picking
   https://github.com/llvm/torch-mlir/pull/844.
3. TODO: Remove the functorch's dependency after make_fx moves to
   pytorch core.
 		2022-05-19 06:18:42 +05:30
..
python_deploy Build each OSX python version in an venv (#852) 2022-05-12 16:39:35 -07:00
torchscript_e2e_heavydep_tests Add simple neural_net and bert_training scripts. 2022-05-19 06:18:42 +05:30
build_python_wheels.sh Pin the Python package to the exact PyTorch nightly. 2022-04-20 16:47:38 -07:00
build_standalone.sh [NFC] Rename external -> externals (#699) 2022-03-26 09:12:27 -07:00
update_shape_lib.sh Introduce new shape library design. 2022-03-15 12:41:58 -07:00
update_torch_ods.sh Reduce compilation time for TorchOps.cpp.inc 2022-03-21 14:42:26 -07:00
write_env_file.sh Move external/torch-mlir to the root of the repo. 2021-09-27 17:11:08 -07:00