mirror of https://github.com/llvm/torch-mlir
c2d3820e48
The current code was inserting all build_list ops after the last constant op since it was assuming that all elements being passed in were constants. This patch replaces that patch with a new function that inserts the build_list ops before the terminator. Also modifies test_export_conv2d_fwd.py since its output no longer matches. TEST: Added test_export_cat.py which is the code in #102 |
||
|---|---|---|
| .. | ||
| c10_dispatch | ||
| type_dispatch | ||
| CMakeLists.txt | ||
| init_python_bindings.cpp | ||
| init_python_bindings.h | ||
| pybind.h | ||