From 4d5c3dc528e68a2266e89b90d03d5842602c1ee1 Mon Sep 17 00:00:00 2001 From: Ze Zhang Date: Fri, 24 Jan 2025 18:35:51 -0800 Subject: [PATCH] [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 --- .clang-format-ignore | 1 + .github/actions/setup-build/action.yml | 1 - deps.bzl | 8 +- lib/Dialect/IR/TcpOps.cpp | 9 +- requirements_lock.txt | 179 ++++++++++++------------- test/Dialect/decompose_tensor_ops.mlir | 7 +- 6 files changed, 100 insertions(+), 105 deletions(-) create mode 100644 .clang-format-ignore diff --git a/.clang-format-ignore b/.clang-format-ignore new file mode 100644 index 00000000..4b22ede2 --- /dev/null +++ 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 index 0c19f7ab..083418b7 100644 --- 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 index 1b9eb7fa..969cce60 100644 --- 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 index fd01fbf5..7dbd2bb7 100644 --- 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 index 8be83636..df15c1df 100644 --- 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 index cc5c8130..c1642a45 100644 --- 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 // CHECK: %[[T0:.*]] = affine.apply #map()[%[[DIM]], %[[DIM_0]]] // CHECK: %[[T1:.*]] = tensor.empty(%[[T0]]) : tensor -// CHECK: %[[DIM_1:.*]] = tensor.dim %[[ARG0]], %[[C0]] : tensor -// CHECK: %[[DIM_2:.*]] = tensor.dim %[[ARG0]], %[[C0]] : tensor -// CHECK: %[[SLICE1:.*]] = tensor.insert_slice %[[ARG0]] into %[[T1]][0, 0] [%[[DIM_2]], 3] [1, 1] : tensor into tensor -// CHECK: %[[DIM_3:.*]] = tensor.dim %[[ARG1]], %[[C0]] : tensor -// CHECK: %[[SLICE_2:.*]] = tensor.insert_slice %arg1 into %[[SLICE1]][%[[DIM_1]], 0] [%[[DIM_3]], 3] [1, 1] : tensor into tensor +// CHECK: %[[SLICE1:.*]] = tensor.insert_slice %[[ARG0]] into %[[T1]][0, 0] [%[[DIM]], 3] [1, 1] : tensor into tensor +// CHECK: %[[SLICE_2:.*]] = tensor.insert_slice %arg1 into %[[SLICE1]][%[[DIM]], 0] [%[[DIM_0]], 3] [1, 1] : tensor into tensor // CHECK: return %[[SLICE_2]] : tensor func.func @tensor_concat_float_tensors(%arg0: tensor, %arg1: tensor) -> tensor { %concat = tensor.concat dim(0) %arg0, %arg1 : (tensor, tensor) -> tensor