Skip to content

Commit 71e9812

Browse files
authored
[LLVM Integrate] Bump llvm, torch-mlir, and stablehlo (#3)
[LLVM](https://github.com/llvm/llvm-project/tree/5d6d982df61d16b6d498e6d59dd91c059679d3d8) (Jan 27) [Torch-MLIR](https://github.com/llvm/torch-mlir/tree/169032010793ee7fe3e305ab920e4119fdfc3b11) (Jan 30) [StableHlo](https://github.com/openxla/stablehlo/tree/b62dc66da9946b4c400c0d99c9d5bb8e04edaee6) (Jan 29) Remove the createFinalizingBufferizePass since it's no longer exist in [upstream](llvm/llvm-project#114154) Add [createConvertControlFlowToLLVMPass](https://github.com/llvm/llvm-project/blob/bd51fa2ad2982811fe76a9d2c0a94073c65e895d/mlir/test/lib/Dialect/LLVM/TestLowerToLLVM.cpp#L81) to lower cf.br and cf.cond_br
1 parent 6b98eeb commit 71e9812

File tree

3 files changed

+36
-32
lines changed

3 files changed

+36
-32
lines changed

deps.bzl

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -22,8 +22,8 @@ def third_party_deps():
2222
path = local_llvm_repo_path(),
2323
)
2424
else:
25-
LLVM_COMMIT = "813f7c3820d00349fe23bfc6ba26159764541540"
26-
LLVM_SHA256 = "a04830a729c030002ce2f1b83e5f894a645fc48830005d40d029aae9acd24340"
25+
LLVM_COMMIT = "5d6d982df61d16b6d498e6d59dd91c059679d3d8"
26+
LLVM_SHA256 = "834184126812eecbdb2ed30de255554a6529295afaf44e9dfd3851d61195dbb5"
2727
http_archive(
2828
name = "llvm-raw",
2929
build_file_content = "# empty",
@@ -39,8 +39,8 @@ def third_party_deps():
3939
path = local_torch_mlir_repo_path(),
4040
)
4141
else:
42-
TORCH_MLIR_COMMIT = "2fb7d6e4dcd06eadb759f2a2e769e53718d0a2f1"
43-
TORCH_MLIR_SHA256 = "111dc77c81706e93728235ffba467277178f10356817e09dc5bd5ab08934323c"
42+
TORCH_MLIR_COMMIT = "169032010793ee7fe3e305ab920e4119fdfc3b11"
43+
TORCH_MLIR_SHA256 = "0f25459b0d6828983c8aa78d139adad4325508bff150b57e97345e9798377dd3"
4444
http_archive(
4545
name = "torch-mlir-raw",
4646
build_file_content = "# empty",
@@ -55,8 +55,8 @@ def third_party_deps():
5555
path = local_stablehlo_repo_path(),
5656
)
5757
else:
58-
STABLEHLO_COMMIT = "6e403b1aa6a71f5eaa09cc720e4ad42f692745e6"
59-
STABLEHLO_SHA256 = "5b6a207f7c41aa986866103a4929e7146acff05830234df5714a45c7100cae3c"
58+
STABLEHLO_COMMIT = "b62dc66da9946b4c400c0d99c9d5bb8e04edaee6"
59+
STABLEHLO_SHA256 = "a51842f5cbcccc2dc74de232793e6fdc0b4403b616281a73bbc704cd227b50db"
6060
http_archive(
6161
name = "stablehlo",
6262
sha256 = STABLEHLO_SHA256,

lib/Pipeline/Pipeline.cpp

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -21,6 +21,7 @@
2121
#include "mlir/Conversion/AffineToStandard/AffineToStandard.h"
2222
#include "mlir/Conversion/ArithToLLVM/ArithToLLVM.h"
2323
#include "mlir/Conversion/BufferizationToMemRef/BufferizationToMemRef.h"
24+
#include "mlir/Conversion/ControlFlowToLLVM/ControlFlowToLLVM.h"
2425
#include "mlir/Conversion/FuncToLLVM/ConvertFuncToLLVMPass.h"
2526
#include "mlir/Conversion/MathToLLVM/MathToLLVM.h"
2627
#include "mlir/Conversion/MathToLibm/MathToLibm.h"
@@ -84,8 +85,6 @@ static void createTcpToLlvmPipeline(OpPassManager &pm) {
8485
bufferizationOptions.setFunctionBoundaryTypeConversion(
8586
bufferization::LayoutMapOption::IdentityLayoutMap);
8687
pm.addPass(bufferization::createOneShotBufferizePass(bufferizationOptions));
87-
pm.addNestedPass<func::FuncOp>(
88-
bufferization::createFinalizingBufferizePass());
8988
// Buffer deallocation pipeline for automatically inserting
9089
// buffer deallocation ops after one-shot bufferization.
9190
// https://sourcegraph.com/github.com/llvm/llvm-project@09bc1e825068f314db71ee7eb32d9f93c5ac87a0/-/blob/mlir/lib/Dialect/Bufferization/Pipelines/BufferizationPipelines.cpp?L21
@@ -122,6 +121,8 @@ static void createTcpToLlvmPipeline(OpPassManager &pm) {
122121
pm.addPass(createFinalizeMemRefToLLVMConversionPass());
123122
// Convert Func to LLVM (always needed).
124123
pm.addPass(createConvertFuncToLLVMPass());
124+
// Convert CF to LLVM (always needed).
125+
pm.addPass(createConvertControlFlowToLLVMPass());
125126

126127
// Sprinkle some cleanups.
127128
pm.addPass(createCanonicalizerPass());

requirements_lock.txt

Lines changed: 27 additions & 24 deletions
Original file line numberDiff line numberDiff line change
@@ -157,31 +157,34 @@ sympy==1.13.1 \
157157
--hash=sha256:9cebf7e04ff162015ce31c9c6c9144daa34a93bd082f54fd8f12deca4f47515f \
158158
--hash=sha256:db36cdc64bf61b9b24578b6f7bab1ecdd2452cf008f34faa33776680c26d66f8
159159
# via torch
160-
torch==2.7.0.dev20250121+cpu \
161-
--hash=sha256:085c327b9d5a2438a79ef62bc03c82910d1285799598d98b940f31f5e391029a \
162-
--hash=sha256:08e96e878f06cef5765e5fdc819262b073d2470241ea73ede337e22c3e7abd69 \
163-
--hash=sha256:098ab0ff55481966f2005cdfeda8a54b34351cd7f56792e0be139a0073f43624 \
164-
--hash=sha256:156142c172dc27efc63be5b0a6fab0150cb0e6dab5d5a6ad0106f293c63db26b \
165-
--hash=sha256:1f2e39a6a3a920e2466bd4668a050f929ab03a6a68f85577b5c9a9d484f7b792 \
166-
--hash=sha256:39be1bc191dc449e017d0aa244100917fb71cca76566440b984b8154281f3000 \
167-
--hash=sha256:41047ab28ebde5bd1713063c3691fcfd40137d2226d42b98620d6a6996fa40e5 \
168-
--hash=sha256:49e2264c58e03bfe58fce3da1179027a29fd03065dfb21e15454c762c556ff14 \
169-
--hash=sha256:527105157e95252d5e30daf6480babbcb1716dbfaefe1eb5d6fe71dca8a078e0 \
170-
--hash=sha256:5ce1adf63bb6b06e1c18c21c14a701a9e0a959a5e93811b1d77f6a60f4fd9641 \
171-
--hash=sha256:61f03785b89ebf54e0edc45105dc65217daead463f72f5eb2b850561a8e4cb5d \
172-
--hash=sha256:8903c69e25dce43bcd033e0e13b31a4dde4c2267b35d15ce1c99cf156b44b07f \
173-
--hash=sha256:91703b06d66f72e00e36f8517639cc5b111f8c56e75c74fec4064475a3a157aa \
174-
--hash=sha256:95973d2de0e29c05554aaed3cad0e0c1cfd56037b3cd047acdc1da55941e8d44 \
175-
--hash=sha256:99799e1a5230fd4ff5b5fea0f38575c9ae59b9d1d0937ef5540d3ff79bd0e7df \
176-
--hash=sha256:a7b42cd2eb41477629ab90a644ed1b1ee44308378854229be59835eff9acea90 \
177-
--hash=sha256:b3345244bd4d19358c6a488a30b02c2b1a63dbbd303a2deed35a339c81b7217e \
178-
--hash=sha256:d58ef0520c38a847c6c887b2cb50e526b628d751023e4c1f02b8b79eb3e0cbb8 \
179-
--hash=sha256:e362a05de22ecb4e43376e58bb094cb89db07e55ffde23c617adc6a0ea50604d
160+
torch==2.7.0.dev20250131+cpu \
161+
--hash=sha256:022eb7ef864640b7b15e4a500dd1354bfc1658cc153e141daf7496d363966179 \
162+
--hash=sha256:1541bffc5543a9b10cd7b35aa9c425ccc46b628f73d4373bef79da6553fe68df \
163+
--hash=sha256:175bf1384ff95f258728b5134c6caec3e6dbaadaa2c458430dd937464295545e \
164+
--hash=sha256:2aea676c6ea0e8a8b7e72ceefe1668cd781a65a58690a9d0e7519a9f5425886e \
165+
--hash=sha256:2b6ddf8dc1040c9ee25984d07ac752a68631b919222d778db94798c24e84d3ff \
166+
--hash=sha256:2f51554094f2d6857ff52517fbb8186b6329a6cacf9fa1de1c0c983542f77678 \
167+
--hash=sha256:3030deb8a2466308f8ebb8b755503735b880442fde88ad93f216f5a866cae5fe \
168+
--hash=sha256:3c5eeaff12c910d9c86b5ebe16f3614abab14744fc961e0c445fb906462f9377 \
169+
--hash=sha256:49c4c50e1557d6b507ea05d762bbf2744c1e9287852baa6d6b75de6c4c011da2 \
170+
--hash=sha256:596a4f0586ad121db78f92b602320a6b581ebde99dd53b056767923c90fc989a \
171+
--hash=sha256:61a5012814824221b3f44552411c60f3b217e0077197288b29646c528abad073 \
172+
--hash=sha256:63fd0a0239f7147f05e840a679904577b7633cf0568494b23aebffca1bea2edf \
173+
--hash=sha256:75a7d8692dd1ac09933fd66980d05c6f13754d19a2bdf9519ad7819b267971cd \
174+
--hash=sha256:859e91760d7151bdf18caf7f2a9afc514c150bda9a4679b1261699dfff56a33d \
175+
--hash=sha256:8e61f9d174ab180592190a011ea0261afdf53f1df85d3eaf7e72a555ea106044 \
176+
--hash=sha256:9f98be38a676681236ac8850b6cece04d53679e95e389952a57dd1fa089f8cb9 \
177+
--hash=sha256:bb9bb80a25c1b4d55ff6e23a016493db31f3b66b43cd87277244cd250e979689 \
178+
--hash=sha256:cb6ca0cdc88cdf80f35925844dcf0c0fd55dcd6d2a7f7a96c20fa47700b38e8b \
179+
--hash=sha256:d3c7633aad4005230783d4e8b87b143b3cef6fdaa2ff695354b4b1670c9b7656 \
180+
--hash=sha256:d5ef26cd431a46792a4ea045995d2900ea2a36a4281d3b67e208ea9988c1caff \
181+
--hash=sha256:f0cf7ab37a4b4e1d38c1b9b2b5cd4a8ae03a46e14707ffc5ee88dfdd6ea9ac41 \
182+
--hash=sha256:f5a9b35bd115f1009064fc36f5d9fff040f9ae096b9a35da09f5293d1b127fe0
180183
# via -r requirements.txt
181-
torch-mlir==20250121.351 \
182-
--hash=sha256:613493ac6c15467e6531ed4f805b57b755085974c2a976bb5d2d929ca4d617f4 \
183-
--hash=sha256:82626b0cfdd1d7ac513960581b3aaf5783c18a62e6684a6ec5de0eaf4efd9b04 \
184-
--hash=sha256:9357bcb06ebb55b3a166b2f85caee52fbd213a0f5e997760ee54401d59aa3bd2
184+
torch-mlir==20250127.357 \
185+
--hash=sha256:43c2362b6a5265405ac5d2291982d6b0d83afafc7ee37165f4cc6b845dec4c15 \
186+
--hash=sha256:62dd44c74212ce772cf245be5610c2ed9cc60c0a51fcf1c3a2e4f1bd3245da29 \
187+
--hash=sha256:b72097a773b3a90adae71a92919868695d32bd487622a32486baa9a75b033bdd
185188
# via -r requirements.txt
186189
typing-extensions==4.12.2 \
187190
--hash=sha256:04e5ca0351e0f3f85c6853954072df659d0d13fac324d0072316b67d7794700d \

0 commit comments

Comments
 (0)