Bump the shape lib to match the upstream functions currently in PyTorch

llvm · Aug 17, 2022 · 121c755 · 121c755
1 parent fde390c
commit 121c755
Show file tree

Hide file tree

Showing 3 changed files with 269 additions and 124 deletions.
diff --git a/build_tools/update_shape_lib.sh b/build_tools/update_shape_lib.sh
@@ -28,5 +28,5 @@ fi
 
 PYTHONPATH="${pypath}" python \
   -m torch_mlir.dialects.torch.importer.jit_ir.build_tools.shape_lib_gen \
-  --pytorch_op_extensions=${ext_module} \
+  --pytorch_op_extensions=${ext_module:-""} \
   --torch_transforms_cpp_dir="${torch_transforms_cpp_dir}"