llvm
/
torch-mlir
mirror of https://github.com/llvm/torch-mlir
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity
torch-mlir/.gitignore

29 lines
287 B
Plaintext
Raw Normal View History

Add vim swap file in gitignore Signed-off-by: Bairen Yi <yibairen.byron@bytedance.com>
2021-02-19 09:17:15 +08:00
*.swp
Add support for constant_pad_nd Note that to enable folding of the code coming from an example like the ConstantPad2dStaticModule e2e test, support for other operations had to be added/improved: - aten::neg.int - aten::eq.float - aten::eq.str - prim::Uninitialized
2022-01-11 15:42:53 +08:00
.cache/
Add stub numpy dialect.
2020-04-27 08:20:58 +08:00
.vscode
Update all python imports to be absolute and use a .env file to set the path correctly. This makes things just work for debugging in VSCode.
2020-05-07 14:25:04 +08:00
.env
NFC: Add .code-workspace to gitignore.
2021-01-09 02:34:49 +08:00
*.code-workspace
Add resnet inference jupyter notebook. This takes the example from torchscript_resnet18_e2e.py and puts it into a slightly cleaned up notebook form. It's still a little rough around the edges. Areas for improvement: - Installation / setup. - API usability. Also, - Add `npcomp-backend-to-iree-frontend-pipeline` since we will be adding more stuff there. - Slight cleanups.
2021-08-10 04:55:20 +08:00
.ipynb_checkpoints
Bump llvm-project to 3d92722f74993969243d1400bc3257ca3d03902f. (#369) * Picks up Python configure changes (was pinned to a bad intermediate commit). * Uses the new mlir_configure_python_dev_packages() to ensure CMake python is found consistently. * Fixes the JIT importer to build as a MODULE vs SHARED (needed for linking to Python as a module, per config changes). * Adds some notes to the README to help folks build a smaller set focused just on this project.
2021-10-22 12:09:00 +08:00
*.venv/
Set some wheel building optimization options. * Also adds a requirements.txt and updates docs to reference it versus stringy pip install. * Adds doc with instructions on creating a wheel. Fixes #370
2021-10-22 12:18:08 +08:00
mlir_venv/
Link against libtorch (#955) This moves torch-mlir to link against libtorch on macOS and linux TESTS: Tests pass. Tested release builds on linux and macOS
2022-07-01 03:40:17 +08:00
externals/pytorch/
libtorch*
Add stub numpy dialect.
2020-04-27 08:20:58 +08:00
Build packages for npcomp-torch. * Adds a minimal setup.py for frontends/pytorch * Makes npcomp-core export its headers and libraries * Adds a script to build packages. * Adds CI step to package and smoke test. * Will need some more tweaks and coordination prior to deploying (version locking etc).
2021-07-30 01:43:18 +08:00
/build/
Add MLIRContext.dense_elements_attr to create an attribute from a python buffer/array.
2020-05-09 08:36:07 +08:00
__pycache__
E2E HuggingFace Bert using LTC Backend (#912) * Update native function definitions * Add ops to support bert lowering - Add empty_strided and as_strided - Restore zeros_like to op blacklist (Without this, tensors will be unintentionally created with a CPU device rather than lazy) - Check for composite implicit ops and add device data IR - Also fix codegen for functionalization * Add autogen to CMakeList * Remove PyTorch submodule * Reduced BERT model size * Print Mark Step status in Torch MLIR LTC debug string * Apply fixes to work with latest upstream/main - Pass importOptions into getMlirTypeFromTorchType during NodeImporter::importNode Without this, the tensor type created may have a mismatched type as ImportOptions may cause vtensor to be used instead of tensor * Update shape inference functions - Fixed compute_shape_native_batch_norm when mean and var are uninitialized Previously, the number of shapes returned would be <3 if either mean or val was didn't exist. Instead, we now initialize them with a vector matching the number of channels. - Implemented compute_shape_mul - Fixed bug in reshape shape inference error message * Get MLIR backend more consistent with TS backend - Remove LazyNativeFunctions::_unsafe_view from autogen - Blacklist ops to make JIT graph more like output of TS backend - Print graph when SSA value has mismatch of types and results - Remove normalize_index from LazyShapeInference - Fix seeds for LTC example models * Update and clean up shape inference functions - Prune shape inference functions - Add shape inference function for GenerateSlice - Add shape inference function for GenerateCopy Co-authored-by: Henry Tu <henry.tu@cerebras.net>
2022-06-08 02:38:50 +08:00
*.pyc
Add MLIRContext.dense_elements_attr to create an attribute from a python buffer/array.
2020-05-09 08:36:07 +08:00
Expand pytype coverage for torch_signature_ods_gen.py
2020-11-25 02:42:56 +08:00
.pytype
Add basic setup.py file for the npcomp-core package. (#256)
2021-07-29 06:58:31 +08:00
# Pip artifacts.
*.egg-info
*.whl
Bring back Python packaging. Will add a CI job that builds and uploads snapshot packages next.
2021-10-06 04:16:12 +08:00
/wheelhouse
Add bazel build support (1/N) (#706) This PR adds rules for building the compiler part with bazel, a followup PRs will build the python bindings.
2022-04-07 02:20:39 +08:00
# Bazel
bazel-*
Torch-MLIR LTC Backend Lowering Codegen (#621) * Codegen and build LTC lowering * Add LazyShapeInference header
2022-02-26 08:50:09 +08:00
Added JIT to MLIR lowering (#724) * Added JIT to MLIR lowering Lowering to JIT is performed in a way similar to how it's done in the TS LTC backend. After a jit::Graph is constructed, it gets converted to a jit::Function, which is fed into the existing utility to generate an MlirModule in torch-mlir. * Renamed `csrc/backend` to `csrc/base_lazy_backend`
2022-04-15 00:40:10 +08:00
# Autogenerated files
Fix LTC Decoupling (#815) * Initial changes * Fix up native functions * Further fix decoupling * Remove unnecessary ops * Formatting and copyright banners: * Add pytorch submodule
2022-05-03 21:35:44 +08:00
/python/torch_mlir/csrc/base_lazy_backend/generated