Skip to content

Commit

Permalink
[LLVM Integrate] Bump llvm, torch-mlir, stablehlo (#36)
Browse files Browse the repository at this point in the history
  • Loading branch information
sjain-stanford committed Feb 1, 2024
1 parent 6b5733f commit b0d474e
Show file tree
Hide file tree
Showing 5 changed files with 36 additions and 6 deletions.
8 changes: 8 additions & 0 deletions .github/workflows/bazelBuildAndTestLlvm.yml
Original file line number Diff line number Diff line change
Expand Up @@ -12,11 +12,13 @@ on:
- main
paths:
- 'deps.bzl'
- '.github/workflows/bazelBuildAndTestLlvm.yml'
push:
branches:
- main
paths:
- 'deps.bzl'
- '.github/workflows/bazelBuildAndTestLlvm.yml'
workflow_dispatch:

# Ensure that only a single job or workflow using the same
Expand Down Expand Up @@ -48,6 +50,12 @@ jobs:
restore-keys: |
llvm-project-bazel-build-cache-${{ runner.os }}
- name: Create bazel cache dir when not found
run: |
if [ ! -d "${HOME}/.cache/bazel" ]; then
mkdir -p "${HOME}/.cache/bazel"
fi
- name: Build docker image
run: |
docker build -f docker/Dockerfile \
Expand Down
8 changes: 8 additions & 0 deletions .github/workflows/bazelBuildAndTestStablehlo.yml
Original file line number Diff line number Diff line change
Expand Up @@ -12,11 +12,13 @@ on:
- main
paths:
- 'deps.bzl'
- '.github/workflows/bazelBuildAndTestStablehlo.yml'
push:
branches:
- main
paths:
- 'deps.bzl'
- '.github/workflows/bazelBuildAndTestStablehlo.yml'
workflow_dispatch:

# Ensure that only a single job or workflow using the same
Expand Down Expand Up @@ -48,6 +50,12 @@ jobs:
restore-keys: |
stablehlo-bazel-build-cache-${{ runner.os }}
- name: Create bazel cache dir when not found
run: |
if [ ! -d "${HOME}/.cache/bazel" ]; then
mkdir -p "${HOME}/.cache/bazel"
fi
- name: Build docker image
run: |
docker build -f docker/Dockerfile \
Expand Down
6 changes: 6 additions & 0 deletions .github/workflows/bazelBuildAndTestTcp.yml
Original file line number Diff line number Diff line change
Expand Up @@ -43,6 +43,12 @@ jobs:
restore-keys: |
mlir-tcp-bazel-build-cache-${{ runner.os }}
- name: Create bazel cache dir when not found
run: |
if [ ! -d "${HOME}/.cache/bazel" ]; then
mkdir -p "${HOME}/.cache/bazel"
fi
- name: Build docker image
run: |
docker build -f docker/Dockerfile \
Expand Down
8 changes: 8 additions & 0 deletions .github/workflows/bazelBuildAndTestTorchmlir.yml
Original file line number Diff line number Diff line change
Expand Up @@ -12,11 +12,13 @@ on:
- main
paths:
- 'deps.bzl'
- '.github/workflows/bazelBuildAndTestTorchmlir.yml'
push:
branches:
- main
paths:
- 'deps.bzl'
- '.github/workflows/bazelBuildAndTestTorchmlir.yml'
workflow_dispatch:

# Ensure that only a single job or workflow using the same
Expand Down Expand Up @@ -48,6 +50,12 @@ jobs:
restore-keys: |
torch-mlir-bazel-build-cache-${{ runner.os }}
- name: Create bazel cache dir when not found
run: |
if [ ! -d "${HOME}/.cache/bazel" ]; then
mkdir -p "${HOME}/.cache/bazel"
fi
- name: Build docker image
run: |
docker build -f docker/Dockerfile \
Expand Down
12 changes: 6 additions & 6 deletions deps.bzl
Original file line number Diff line number Diff line change
Expand Up @@ -22,8 +22,8 @@ def third_party_deps():
path = local_llvm_repo_path(),
)
else:
LLVM_COMMIT = "6b65d79fbb4682468333cea42b62f15c2dffd8f3"
LLVM_SHA256 = "99278b9422998e248703c348565ca973b10678b65697dcf4dcc80acb9c174c2a"
LLVM_COMMIT = "70eb0e37a86747f9266e4c8380baa89746f5e23b"
LLVM_SHA256 = "a96a6cad1db1726b757df136d33328e49ee8a215e824d7bdcadb09612522001a"
http_archive(
name = "llvm-raw",
build_file_content = "# empty",
Expand All @@ -39,8 +39,8 @@ def third_party_deps():
path = local_torch_mlir_repo_path(),
)
else:
TORCH_MLIR_COMMIT = "670a99ae196da892310776f110cfe29dfb68a174"
TORCH_MLIR_SHA256 = "7bec6b45d848718e255a3f59fe41a96126db25df84e69b3974d81f4ac0d0a65e"
TORCH_MLIR_COMMIT = "c7d7d7f00494b588c31ac617e91354b12709009d"
TORCH_MLIR_SHA256 = "0992e854412462fa8c0cd59c0f0cf0d76b3d0cb28bec51e277073f7f8beaed86"
http_archive(
name = "torch-mlir-raw",
sha256 = TORCH_MLIR_SHA256,
Expand All @@ -55,8 +55,8 @@ def third_party_deps():
path = local_stablehlo_repo_path(),
)
else:
STABLEHLO_COMMIT = "3260a31f09744419377dae409043f12bb7418c38"
STABLEHLO_SHA256 = "41cacc34cb591fbc73674f135475928987a1e8ff929ad1329b17bd75a449d102"
STABLEHLO_COMMIT = "fd52182f76cadb82f2064fe5fc49a4fb4347a826"
STABLEHLO_SHA256 = "f4f79d7e59650cfff4fd9d3d127e70f5d5b9973fce6a10e5964facb029ebf245"
http_archive(
name = "stablehlo",
sha256 = STABLEHLO_SHA256,
Expand Down

0 comments on commit b0d474e

Please sign in to comment.