[LLVM Integrate] Bump llvm, torch-mlir (#105)

[LLVM](https://github.com/llvm/llvm-project/tree/813f7c3820d00349fe23bfc6ba26159764541540) (Nov 13) [Torch-MLIR](https://github.com/llvm/torch-mlir/tree/2fb7d6e4dcd06eadb759f2a2e769e53718d0a2f1) (Jan 21) [StableHlo](https://github.com/openxla/stablehlo/tree/6e403b1aa6a71f5eaa09cc720e4ad42f692745e6) (Oct 25 - No Change) Also fix a lit test
cruise-automation · Jan 25, 2025 · 4d5c3dc · 4d5c3dc
1 parent bf8f06e
commit 4d5c3dc
Show file tree

Hide file tree

Showing 6 changed files with 100 additions and 105 deletions.
diff --git a/.clang-format-ignore b/.clang-format-ignore
@@ -0,0 +1 @@
+tools/aot/execute_test.template.cpp
diff --git a/.github/actions/setup-build/action.yml b/.github/actions/setup-build/action.yml
@@ -41,7 +41,6 @@ runs:
         sudo apt-get remove -y google-cloud-cli --fix-missing || echo "::debug::The command [sudo apt-get remove -y google-cloud-cli --fix-missing] failed to complete successfully. Proceeding..."
         sudo apt-get autoremove -y || echo "::warning::The command [sudo apt-get autoremove -y] failed to complete successfully. Proceeding..."
         sudo apt-get clean || echo "::warning::The command [sudo apt-get clean] failed to complete successfully. Proceeding..."
-        docker rmi $(docker image ls -aq)
         sudo docker image prune --all --force || true
         df -h
 

diff --git a/deps.bzl b/deps.bzl
@@ -22,8 +22,8 @@ def third_party_deps():
             path = local_llvm_repo_path(),
         )
     else:
-        LLVM_COMMIT = "6c64c8a6f3f77c30745c751d4163ff6bf2fc323b"
-        LLVM_SHA256 = "a27f8452e8a4267acb7bf59ea4dae012cec22e37624a67206b927985554e3640"
+        LLVM_COMMIT = "813f7c3820d00349fe23bfc6ba26159764541540"
+        LLVM_SHA256 = "a04830a729c030002ce2f1b83e5f894a645fc48830005d40d029aae9acd24340"
         http_archive(
             name = "llvm-raw",
             build_file_content = "# empty",
@@ -39,8 +39,8 @@ def third_party_deps():
             path = local_torch_mlir_repo_path(),
         )
     else:
-        TORCH_MLIR_COMMIT = "30c519369ed7eabad0282d0f874500a9b41fcbbd"
-        TORCH_MLIR_SHA256 = "9598ab5bb4a75cd63a078fccb19b380f753906153a92ff7ee5dca3eba93deba5"
+        TORCH_MLIR_COMMIT = "2fb7d6e4dcd06eadb759f2a2e769e53718d0a2f1"
+        TORCH_MLIR_SHA256 = "111dc77c81706e93728235ffba467277178f10356817e09dc5bd5ab08934323c"
         http_archive(
             name = "torch-mlir-raw",
             build_file_content = "# empty",

diff --git a/lib/Dialect/IR/TcpOps.cpp b/lib/Dialect/IR/TcpOps.cpp
@@ -192,9 +192,9 @@ LogicalResult GatherOp::verify() {
           i == gatherDim)) {
       std::stringstream ss;
       ss << "indicies index " << i
-         << " expected to be less than or equal to input "
-         << " (" << indicesTensor.getShape()[i]
-         << " <= " << inputTensor.getShape()[i] << ")";
+         << " expected to be less than or equal to input " << " ("
+         << indicesTensor.getShape()[i] << " <= " << inputTensor.getShape()[i]
+         << ")";
       return emitOpError(ss.str());
     }
   }
@@ -293,8 +293,7 @@ ParseResult BindSymbolicShapeOp::parse(OpAsmParser &parser,
 void BindSymbolicShapeOp::print(OpAsmPrinter &p) {
   p << " " << getOperand() << ", [";
   llvm::interleaveComma(getShapeSymbols(), p);
-  p << "], "
-    << "affine_map<" << getShapeExpressions().getValue() << ">";
+  p << "], " << "affine_map<" << getShapeExpressions().getValue() << ">";
   p.printOptionalAttrDict((*this)->getAttrs(),
                           /*elidedAttrs=*/{"shape_expressions"});
   p << " : " << getOperand().getType();

diff --git a/requirements_lock.txt b/requirements_lock.txt
@@ -7,17 +7,17 @@
 --find-links https://github.com/llvm/torch-mlir-release/releases/expanded_assets/dev-wheels
 --find-links https://download.pytorch.org/whl/nightly/cpu/torch/
 
-filelock==3.16.1 \
-    --hash=sha256:2082e5703d51fbf98ea75855d9d5527e33d8ff23099bec374a134febee6946b0 \
-    --hash=sha256:c249fbfcd5db47e5e2d6d62198e565475ee65e4831e2561c8e313fa7eb961435
+filelock==3.17.0 \
+    --hash=sha256:533dc2f7ba78dc2f0f531fc6c4940addf7b70a481e269a5a3b93be94ffbe8338 \
+    --hash=sha256:ee4e77401ef576ebb38cd7f13b9b28893194acc20a8e68e18730ba9c0e54660e
     # via torch
-fsspec==2024.10.0 \
-    --hash=sha256:03b9a6785766a4de40368b88906366755e2819e758b83705c88cd7cb5fe81871 \
-    --hash=sha256:eda2d8a4116d4f2429db8550f2457da57279247dd930bb12f821b58391359493
+fsspec==2024.12.0 \
+    --hash=sha256:670700c977ed2fb51e0d9f9253177ed20cbde4a3e5c0283cc5385b5870c8533f \
+    --hash=sha256:b520aed47ad9804237ff878b504267a3b0b441e97508bd6d2d8774e3db85cee2
     # via torch
-jinja2==3.1.4 \
-    --hash=sha256:4a3aee7acbbe7303aede8e9648d13b8bf88a429282aa6122a993f0ac800cb369 \
-    --hash=sha256:bc5dd2abb727a5319567b7a813e6a2e7318c39f4f487cfe6c89c6f9c7d25197d
+jinja2==3.1.5 \
+    --hash=sha256:8fefff8dc3034e27bb80d67c671eb8a9bc424c0ef4c0826edbff304cceff43bb \
+    --hash=sha256:aba0f4dc9ed8013c424088f68a5c226f7d6097ed89b246d7749c2ec4175c6adb
     # via torch
 markupsafe==3.0.2 \
     --hash=sha256:0bff5e0ae4ef2e1ae4fdf2dfd5b76c75e5c2fa4132d05fc1b0dabcd20c7e28c4 \
@@ -90,62 +90,62 @@ networkx==3.4.2 \
     --hash=sha256:307c3669428c5362aab27c8a1260aa8f47c4e91d3891f48be0141738d8d053e1 \
     --hash=sha256:df5d4365b724cf81b8c6a7312509d0c22386097011ad1abe274afd5e9d3bbc5f
     # via torch
-numpy==2.2.0 \
-    --hash=sha256:0557eebc699c1c34cccdd8c3778c9294e8196df27d713706895edc6f57d29608 \
-    --hash=sha256:0798b138c291d792f8ea40fe3768610f3c7dd2574389e37c3f26573757c8f7ef \
-    --hash=sha256:0da8495970f6b101ddd0c38ace92edea30e7e12b9a926b57f5fabb1ecc25bb90 \
-    --hash=sha256:0f0986e917aca18f7a567b812ef7ca9391288e2acb7a4308aa9d265bd724bdae \
-    --hash=sha256:122fd2fcfafdefc889c64ad99c228d5a1f9692c3a83f56c292618a59aa60ae83 \
-    --hash=sha256:140dd80ff8981a583a60980be1a655068f8adebf7a45a06a6858c873fcdcd4a0 \
-    --hash=sha256:16757cf28621e43e252c560d25b15f18a2f11da94fea344bf26c599b9cf54b73 \
-    --hash=sha256:18142b497d70a34b01642b9feabb70156311b326fdddd875a9981f34a369b671 \
-    --hash=sha256:1c92113619f7b272838b8d6702a7f8ebe5edea0df48166c47929611d0b4dea69 \
-    --hash=sha256:1e25507d85da11ff5066269d0bd25d06e0a0f2e908415534f3e603d2a78e4ffa \
-    --hash=sha256:30bf971c12e4365153afb31fc73f441d4da157153f3400b82db32d04de1e4066 \
-    --hash=sha256:3579eaeb5e07f3ded59298ce22b65f877a86ba8e9fe701f5576c99bb17c283da \
-    --hash=sha256:36b2b43146f646642b425dd2027730f99bac962618ec2052932157e213a040e9 \
-    --hash=sha256:3905a5fffcc23e597ee4d9fb3fcd209bd658c352657548db7316e810ca80458e \
-    --hash=sha256:3a4199f519e57d517ebd48cb76b36c82da0360781c6a0353e64c0cac30ecaad3 \
-    --hash=sha256:3f2f5cddeaa4424a0a118924b988746db6ffa8565e5829b1841a8a3bd73eb59a \
-    --hash=sha256:40deb10198bbaa531509aad0cd2f9fadb26c8b94070831e2208e7df543562b74 \
-    --hash=sha256:440cfb3db4c5029775803794f8638fbdbf71ec702caf32735f53b008e1eaece3 \
-    --hash=sha256:4723a50e1523e1de4fccd1b9a6dcea750c2102461e9a02b2ac55ffeae09a4410 \
-    --hash=sha256:4bddbaa30d78c86329b26bd6aaaea06b1e47444da99eddac7bf1e2fab717bd72 \
-    --hash=sha256:4e58666988605e251d42c2818c7d3d8991555381be26399303053b58a5bbf30d \
-    --hash=sha256:54dc1d6d66f8d37843ed281773c7174f03bf7ad826523f73435deb88ba60d2d4 \
-    --hash=sha256:57fcc997ffc0bef234b8875a54d4058afa92b0b0c4223fc1f62f24b3b5e86038 \
-    --hash=sha256:58b92a5828bd4d9aa0952492b7de803135038de47343b2aa3cc23f3b71a3dc4e \
-    --hash=sha256:5a145e956b374e72ad1dff82779177d4a3c62bc8248f41b80cb5122e68f22d13 \
-    --hash=sha256:6ab153263a7c5ccaf6dfe7e53447b74f77789f28ecb278c3b5d49db7ece10d6d \
-    --hash=sha256:7832f9e8eb00be32f15fdfb9a981d6955ea9adc8574c521d48710171b6c55e95 \
-    --hash=sha256:7fe4bb0695fe986a9e4deec3b6857003b4cfe5c5e4aac0b95f6a658c14635e31 \
-    --hash=sha256:7fe8f3583e0607ad4e43a954e35c1748b553bfe9fdac8635c02058023277d1b3 \
-    --hash=sha256:85ad7d11b309bd132d74397fcf2920933c9d1dc865487128f5c03d580f2c3d03 \
-    --hash=sha256:9874bc2ff574c40ab7a5cbb7464bf9b045d617e36754a7bc93f933d52bd9ffc6 \
-    --hash=sha256:a184288538e6ad699cbe6b24859206e38ce5fba28f3bcfa51c90d0502c1582b2 \
-    --hash=sha256:a222d764352c773aa5ebde02dd84dba3279c81c6db2e482d62a3fa54e5ece69b \
-    --hash=sha256:a50aeff71d0f97b6450d33940c7181b08be1441c6c193e678211bff11aa725e7 \
-    --hash=sha256:a55dc7a7f0b6198b07ec0cd445fbb98b05234e8b00c5ac4874a63372ba98d4ab \
-    --hash=sha256:a62eb442011776e4036af5c8b1a00b706c5bc02dc15eb5344b0c750428c94219 \
-    --hash=sha256:a7d41d1612c1a82b64697e894b75db6758d4f21c3ec069d841e60ebe54b5b571 \
-    --hash=sha256:a98f6f20465e7618c83252c02041517bd2f7ea29be5378f09667a8f654a5918d \
-    --hash=sha256:afe8fb968743d40435c3827632fd36c5fbde633b0423da7692e426529b1759b1 \
-    --hash=sha256:b0b227dcff8cdc3efbce66d4e50891f04d0a387cce282fe1e66199146a6a8fca \
-    --hash=sha256:b30042fe92dbd79f1ba7f6898fada10bdaad1847c44f2dff9a16147e00a93661 \
-    --hash=sha256:b606b1aaf802e6468c2608c65ff7ece53eae1a6874b3765f69b8ceb20c5fa78e \
-    --hash=sha256:b6207dc8fb3c8cb5668e885cef9ec7f70189bec4e276f0ff70d5aa078d32c88e \
-    --hash=sha256:c2aed8fcf8abc3020d6a9ccb31dbc9e7d7819c56a348cc88fd44be269b37427e \
-    --hash=sha256:cb24cca1968b21355cc6f3da1a20cd1cebd8a023e3c5b09b432444617949085a \
-    --hash=sha256:cff210198bb4cae3f3c100444c5eaa573a823f05c253e7188e1362a5555235b3 \
-    --hash=sha256:d35717333b39d1b6bb8433fa758a55f1081543de527171543a2b710551d40881 \
-    --hash=sha256:df12a1f99b99f569a7c2ae59aa2d31724e8d835fc7f33e14f4792e3071d11221 \
-    --hash=sha256:e09d40edfdb4e260cb1567d8ae770ccf3b8b7e9f0d9b5c2a9992696b30ce2742 \
-    --hash=sha256:e12c6c1ce84628c52d6367863773f7c8c8241be554e8b79686e91a43f1733773 \
-    --hash=sha256:e2b8cd48a9942ed3f85b95ca4105c45758438c7ed28fff1e4ce3e57c3b589d8e \
-    --hash=sha256:e500aba968a48e9019e42c0c199b7ec0696a97fa69037bea163b55398e390529 \
-    --hash=sha256:ebe5e59545401fbb1b24da76f006ab19734ae71e703cdb4a8b347e84a0cece67 \
-    --hash=sha256:f0dd071b95bbca244f4cb7f70b77d2ff3aaaba7fa16dc41f58d14854a6204e6c \
-    --hash=sha256:f8c8b141ef9699ae777c6278b52c706b653bf15d135d302754f6b2e90eb30367
+numpy==2.2.2 \
+    --hash=sha256:02935e2c3c0c6cbe9c7955a8efa8908dd4221d7755644c59d1bba28b94fd334f \
+    --hash=sha256:0349b025e15ea9d05c3d63f9657707a4e1d471128a3b1d876c095f328f8ff7f0 \
+    --hash=sha256:09d6a2032faf25e8d0cadde7fd6145118ac55d2740132c1d845f98721b5ebcfd \
+    --hash=sha256:0bc61b307655d1a7f9f4b043628b9f2b721e80839914ede634e3d485913e1fb2 \
+    --hash=sha256:0eec19f8af947a61e968d5429f0bd92fec46d92b0008d0a6685b40d6adf8a4f4 \
+    --hash=sha256:106397dbbb1896f99e044efc90360d098b3335060375c26aa89c0d8a97c5f648 \
+    --hash=sha256:128c41c085cab8a85dc29e66ed88c05613dccf6bc28b3866cd16050a2f5448be \
+    --hash=sha256:149d1113ac15005652e8d0d3f6fd599360e1a708a4f98e43c9c77834a28238cb \
+    --hash=sha256:159ff6ee4c4a36a23fe01b7c3d07bd8c14cc433d9720f977fcd52c13c0098160 \
+    --hash=sha256:22ea3bb552ade325530e72a0c557cdf2dea8914d3a5e1fecf58fa5dbcc6f43cd \
+    --hash=sha256:23ae9f0c2d889b7b2d88a3791f6c09e2ef827c2446f1c4a3e3e76328ee4afd9a \
+    --hash=sha256:250c16b277e3b809ac20d1f590716597481061b514223c7badb7a0f9993c7f84 \
+    --hash=sha256:2ec6c689c61df613b783aeb21f945c4cbe6c51c28cb70aae8430577ab39f163e \
+    --hash=sha256:2ffbb1acd69fdf8e89dd60ef6182ca90a743620957afb7066385a7bbe88dc748 \
+    --hash=sha256:3074634ea4d6df66be04f6728ee1d173cfded75d002c75fac79503a880bf3825 \
+    --hash=sha256:356ca982c188acbfa6af0d694284d8cf20e95b1c3d0aefa8929376fea9146f60 \
+    --hash=sha256:3fbe72d347fbc59f94124125e73fc4976a06927ebc503ec5afbfb35f193cd957 \
+    --hash=sha256:40c7ff5da22cd391944a28c6a9c638a5eef77fcf71d6e3a79e1d9d9e82752715 \
+    --hash=sha256:41184c416143defa34cc8eb9d070b0a5ba4f13a0fa96a709e20584638254b317 \
+    --hash=sha256:451e854cfae0febe723077bd0cf0a4302a5d84ff25f0bfece8f29206c7bed02e \
+    --hash=sha256:4525b88c11906d5ab1b0ec1f290996c0020dd318af8b49acaa46f198b1ffc283 \
+    --hash=sha256:463247edcee4a5537841d5350bc87fe8e92d7dd0e8c71c995d2c6eecb8208278 \
+    --hash=sha256:4dbd80e453bd34bd003b16bd802fac70ad76bd463f81f0c518d1245b1c55e3d9 \
+    --hash=sha256:57b4012e04cc12b78590a334907e01b3a85efb2107df2b8733ff1ed05fce71de \
+    --hash=sha256:5a8c863ceacae696aff37d1fd636121f1a512117652e5dfb86031c8d84836369 \
+    --hash=sha256:5acea83b801e98541619af398cc0109ff48016955cc0818f478ee9ef1c5c3dcb \
+    --hash=sha256:642199e98af1bd2b6aeb8ecf726972d238c9877b0f6e8221ee5ab945ec8a2189 \
+    --hash=sha256:64bd6e1762cd7f0986a740fee4dff927b9ec2c5e4d9a28d056eb17d332158014 \
+    --hash=sha256:6d9fc9d812c81e6168b6d405bf00b8d6739a7f72ef22a9214c4241e0dc70b323 \
+    --hash=sha256:7079129b64cb78bdc8d611d1fd7e8002c0a2565da6a47c4df8062349fee90e3e \
+    --hash=sha256:7dca87ca328f5ea7dafc907c5ec100d187911f94825f8700caac0b3f4c384b49 \
+    --hash=sha256:860fd59990c37c3ef913c3ae390b3929d005243acca1a86facb0773e2d8d9e50 \
+    --hash=sha256:8e6da5cffbbe571f93588f562ed130ea63ee206d12851b60819512dd3e1ba50d \
+    --hash=sha256:8ec0636d3f7d68520afc6ac2dc4b8341ddb725039de042faf0e311599f54eb37 \
+    --hash=sha256:9491100aba630910489c1d0158034e1c9a6546f0b1340f716d522dc103788e39 \
+    --hash=sha256:97b974d3ba0fb4612b77ed35d7627490e8e3dff56ab41454d9e8b23448940576 \
+    --hash=sha256:995f9e8181723852ca458e22de5d9b7d3ba4da3f11cc1cb113f093b271d7965a \
+    --hash=sha256:9dd47ff0cb2a656ad69c38da850df3454da88ee9a6fde0ba79acceee0e79daba \
+    --hash=sha256:9fad446ad0bc886855ddf5909cbf8cb5d0faa637aaa6277fb4b19ade134ab3c7 \
+    --hash=sha256:a972cec723e0563aa0823ee2ab1df0cb196ed0778f173b381c871a03719d4826 \
+    --hash=sha256:ac9bea18d6d58a995fac1b2cb4488e17eceeac413af014b1dd26170b766d8467 \
+    --hash=sha256:b0531f0b0e07643eb089df4c509d30d72c9ef40defa53e41363eca8a8cc61495 \
+    --hash=sha256:b208cfd4f5fe34e1535c08983a1a6803fdbc7a1e86cf13dd0c61de0b51a0aadc \
+    --hash=sha256:b3482cb7b3325faa5f6bc179649406058253d91ceda359c104dac0ad320e1391 \
+    --hash=sha256:b6fb9c32a91ec32a689ec6410def76443e3c750e7cfc3fb2206b985ffb2b85f0 \
+    --hash=sha256:b78ea78450fd96a498f50ee096f69c75379af5138f7881a51355ab0e11286c97 \
+    --hash=sha256:bd249bc894af67cbd8bad2c22e7cbcd46cf87ddfca1f1289d1e7e54868cc785c \
+    --hash=sha256:c7d1fd447e33ee20c1f33f2c8e6634211124a9aabde3c617687d8b739aa69eac \
+    --hash=sha256:d0bbe7dd86dca64854f4b6ce2ea5c60b51e36dfd597300057cf473d3615f2369 \
+    --hash=sha256:d6d6a0910c3b4368d89dde073e630882cdb266755565155bc33520283b2d9df8 \
+    --hash=sha256:da1eeb460ecce8d5b8608826595c777728cdf28ce7b5a5a8c8ac8d949beadcf2 \
+    --hash=sha256:e0c8854b09bc4de7b041148d8550d3bd712b5c21ff6a8ed308085f190235d7ff \
+    --hash=sha256:e0d4142eb40ca6f94539e4db929410f2a46052a0fe7a2c1c59f6179c39938d2a \
+    --hash=sha256:e9e82dcb3f2ebbc8cb5ce1102d5f1c5ed236bf8a11730fb45ba82e2841ec21df \
+    --hash=sha256:ed6906f61834d687738d25988ae117683705636936cc605be0bb208b23df4d8f
     # via
     #   -r requirements.txt
     #   torch-mlir
@@ -157,32 +157,31 @@ sympy==1.13.1 \
     --hash=sha256:9cebf7e04ff162015ce31c9c6c9144daa34a93bd082f54fd8f12deca4f47515f \
     --hash=sha256:db36cdc64bf61b9b24578b6f7bab1ecdd2452cf008f34faa33776680c26d66f8
     # via torch
-torch==2.6.0.dev20241213+cpu \
-    --hash=sha256:017e4bf45c896ac21e17620efaa43cab89eebbb710a65949db865395ae6fa4b3 \
-    --hash=sha256:05d8e66b3bdc16516da6fd9745ba490f2c4e99d08d1ed1c5cbf1b345b5433bf1 \
-    --hash=sha256:097173fd80be1cf4a59096e966652350b76b6fd288193f133e088c286fc4a908 \
-    --hash=sha256:247514631d11bac9db1f0720eacd1add7ff999cfff68ecfeb56ffea5cb9edc82 \
-    --hash=sha256:28018de507674792d042b71682e08bec1a3983dfcfca8cf3fb7d786181fa74d8 \
-    --hash=sha256:290c7119de3f4111e50508f4f6f96bc5e82d38f3fda40ece4dc889a802ede05a \
-    --hash=sha256:2f6b27bedec3538769cadbe7896aacb2180aa64d3d1172ab506a642e14c1e0fe \
-    --hash=sha256:41c833ab08fdee8cf26e26a1ad50947a0f15d66cd5bbd2230b0fdf5d92c43169 \
-    --hash=sha256:4c62e8985c73d299da217ecc486a7c32c4e28fb528f3fc0be45b0d78c67f7274 \
-    --hash=sha256:4d512fbe7be28164e3686d746cbe1b12b8b2bc0a0a3ff400d0d69664f2a0f946 \
-    --hash=sha256:5e4b1bcb60cf0af7fe965a016dd04e725ca9003ba30e083b7c365f8b7f6b8c72 \
-    --hash=sha256:66091621133071ac2467e11ed8195df3449cffad3bce8195059053e5d2b75f2a \
-    --hash=sha256:7faf09eb3ec565ffe81c02958255e7538bf0bdeffe92aa1a3fae36dcea66abc5 \
-    --hash=sha256:864189591e66345b170e7c620c583f773b057c19d5023a2053bce44cd965f529 \
-    --hash=sha256:8e1870b0ca0056c29aa4a19f8576a3f843333cfeaa46f9dadc7878e856db1bd4 \
-    --hash=sha256:95e7fc22407c8a5ae995e6c22015909d0297c6422e03dbbdbd2fd007b618c926 \
-    --hash=sha256:99c368e7352823a59c9a89f099150488914fa9cef7fd652413d229baf34202a3 \
-    --hash=sha256:c16b4cae2e16e10c28df72dcb745ea8813079057ba711ab3196c3c88af6f61b3 \
-    --hash=sha256:d50edfa13fa00bef851a5e3c8f864d895a83d1abe737cd6e9905696ac7b6796e \
-    --hash=sha256:ddcefe724067e9937bf39bb606b954ffe1c96ff2a6f4301f5a12210079944dca \
-    --hash=sha256:f9219072e2c8b701172a602e1106b383cf7cc6a291d8a4b370f2ea768ec6e2de
+torch==2.7.0.dev20250121+cpu \
+    --hash=sha256:085c327b9d5a2438a79ef62bc03c82910d1285799598d98b940f31f5e391029a \
+    --hash=sha256:08e96e878f06cef5765e5fdc819262b073d2470241ea73ede337e22c3e7abd69 \
+    --hash=sha256:098ab0ff55481966f2005cdfeda8a54b34351cd7f56792e0be139a0073f43624 \
+    --hash=sha256:156142c172dc27efc63be5b0a6fab0150cb0e6dab5d5a6ad0106f293c63db26b \
+    --hash=sha256:1f2e39a6a3a920e2466bd4668a050f929ab03a6a68f85577b5c9a9d484f7b792 \
+    --hash=sha256:39be1bc191dc449e017d0aa244100917fb71cca76566440b984b8154281f3000 \
+    --hash=sha256:41047ab28ebde5bd1713063c3691fcfd40137d2226d42b98620d6a6996fa40e5 \
+    --hash=sha256:49e2264c58e03bfe58fce3da1179027a29fd03065dfb21e15454c762c556ff14 \
+    --hash=sha256:527105157e95252d5e30daf6480babbcb1716dbfaefe1eb5d6fe71dca8a078e0 \
+    --hash=sha256:5ce1adf63bb6b06e1c18c21c14a701a9e0a959a5e93811b1d77f6a60f4fd9641 \
+    --hash=sha256:61f03785b89ebf54e0edc45105dc65217daead463f72f5eb2b850561a8e4cb5d \
+    --hash=sha256:8903c69e25dce43bcd033e0e13b31a4dde4c2267b35d15ce1c99cf156b44b07f \
+    --hash=sha256:91703b06d66f72e00e36f8517639cc5b111f8c56e75c74fec4064475a3a157aa \
+    --hash=sha256:95973d2de0e29c05554aaed3cad0e0c1cfd56037b3cd047acdc1da55941e8d44 \
+    --hash=sha256:99799e1a5230fd4ff5b5fea0f38575c9ae59b9d1d0937ef5540d3ff79bd0e7df \
+    --hash=sha256:a7b42cd2eb41477629ab90a644ed1b1ee44308378854229be59835eff9acea90 \
+    --hash=sha256:b3345244bd4d19358c6a488a30b02c2b1a63dbbd303a2deed35a339c81b7217e \
+    --hash=sha256:d58ef0520c38a847c6c887b2cb50e526b628d751023e4c1f02b8b79eb3e0cbb8 \
+    --hash=sha256:e362a05de22ecb4e43376e58bb094cb89db07e55ffde23c617adc6a0ea50604d
     # via -r requirements.txt
-torch-mlir==20241213.312 \
-    --hash=sha256:5d2db558bcbade39d68eb621aa0d9ec749d984b0c47ba5171faf56522dc9bbf6 \
-    --hash=sha256:c255f68e584fee66767120949d76020d1cbf97d50bb5309950123593cf78e941
+torch-mlir==20250121.351 \
+    --hash=sha256:613493ac6c15467e6531ed4f805b57b755085974c2a976bb5d2d929ca4d617f4 \
+    --hash=sha256:82626b0cfdd1d7ac513960581b3aaf5783c18a62e6684a6ec5de0eaf4efd9b04 \
+    --hash=sha256:9357bcb06ebb55b3a166b2f85caee52fbd213a0f5e997760ee54401d59aa3bd2
     # via -r requirements.txt
 typing-extensions==4.12.2 \
     --hash=sha256:04e5ca0351e0f3f85c6853954072df659d0d13fac324d0072316b67d7794700d \

diff --git a/test/Dialect/decompose_tensor_ops.mlir b/test/Dialect/decompose_tensor_ops.mlir
@@ -8,11 +8,8 @@
 // CHECK:        %[[DIM_0:.*]] = tensor.dim %[[ARG1]], %[[C0]] : tensor<?x3xf32>
 // CHECK:        %[[T0:.*]] = affine.apply #map()[%[[DIM]], %[[DIM_0]]]
 // CHECK:        %[[T1:.*]] = tensor.empty(%[[T0]]) : tensor<?x3xf32>
-// CHECK:        %[[DIM_1:.*]] = tensor.dim %[[ARG0]], %[[C0]] : tensor<?x3xf32>
-// CHECK:        %[[DIM_2:.*]] = tensor.dim %[[ARG0]], %[[C0]] : tensor<?x3xf32>
-// CHECK:        %[[SLICE1:.*]] = tensor.insert_slice %[[ARG0]] into %[[T1]][0, 0] [%[[DIM_2]], 3] [1, 1] : tensor<?x3xf32> into tensor<?x3xf32>
-// CHECK:        %[[DIM_3:.*]] = tensor.dim %[[ARG1]], %[[C0]] : tensor<?x3xf32>
-// CHECK:        %[[SLICE_2:.*]] = tensor.insert_slice %arg1 into %[[SLICE1]][%[[DIM_1]], 0] [%[[DIM_3]], 3] [1, 1] : tensor<?x3xf32> into tensor<?x3xf32>
+// CHECK:        %[[SLICE1:.*]] = tensor.insert_slice %[[ARG0]] into %[[T1]][0, 0] [%[[DIM]], 3] [1, 1] : tensor<?x3xf32> into tensor<?x3xf32>
+// CHECK:        %[[SLICE_2:.*]] = tensor.insert_slice %arg1 into %[[SLICE1]][%[[DIM]], 0] [%[[DIM_0]], 3] [1, 1] : tensor<?x3xf32> into tensor<?x3xf32>
 // CHECK:        return %[[SLICE_2]] : tensor<?x3xf32>
 func.func @tensor_concat_float_tensors(%arg0: tensor<?x3xf32>, %arg1: tensor<?x3xf32>) -> tensor<?x3xf32> {                                                                                                                                                                             
   %concat = tensor.concat dim(0) %arg0, %arg1 : (tensor<?x3xf32>, tensor<?x3xf32>) -> tensor<?x3xf32>