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 |
||
|---|---|---|
| .. | ||
| CMakeLists.txt | ||
| acap_dispatch.cpp | ||
| acap_dispatch.h | ||
| debug.cpp | ||
| debug.h | ||
| func_builder.cpp | ||
| func_builder.h | ||
| module_builder.cpp | ||
| module_builder.h | ||
| python_bindings.cpp | ||