Skip to content

Commit

Permalink
Merge branch 'main' into reranker-python
Browse files Browse the repository at this point in the history
  • Loading branch information
kaixuanliu authored Dec 17, 2024
2 parents bcc20d3 + 57d8fc8 commit a28ad28
Show file tree
Hide file tree
Showing 67 changed files with 36,411 additions and 1,686 deletions.
2 changes: 2 additions & 0 deletions .github/workflows/build.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -127,6 +127,7 @@ jobs:

- name: Extract metadata (tags, labels) for Docker
id: meta-grpc
if: ${{ matrix.grpc }}
uses: docker/metadata-action@v5
with:
images: |
Expand All @@ -142,6 +143,7 @@ jobs:
- name: Build and push Docker image
id: build-and-push-grpc
if: ${{ matrix.grpc }}
uses: docker/build-push-action@v6
with:
context: .
Expand Down
19 changes: 13 additions & 6 deletions .github/workflows/matrix.json
Original file line number Diff line number Diff line change
Expand Up @@ -2,10 +2,11 @@
{
"name": "turing",
"imageNamePrefix": "turing-",
"runOn": "main",
"runOn": "always",
"sccache": true,
"cudaComputeCap": 75,
"extraBuildArgs": "DEFAULT_USE_FLASH_ATTENTION=False",
"grpc": true,
"dockerfile": "Dockerfile-cuda"
},
{
Expand All @@ -14,44 +15,50 @@
"runOn": "always",
"sccache": true,
"cudaComputeCap": 80,
"grpc": true,
"dockerfile": "Dockerfile-cuda"
},
{
"name": "a10",
"imageNamePrefix": "86-",
"runOn": "main",
"runOn": "always",
"sccache": true,
"cudaComputeCap": 86,
"grpc": true,
"dockerfile": "Dockerfile-cuda"
},
{
"name": "RTX 4000",
"imageNamePrefix": "89-",
"runOn": "main",
"runOn": "always",
"sccache": true,
"cudaComputeCap": 89,
"grpc": true,
"dockerfile": "Dockerfile-cuda"
},
{
"name": "Hopper",
"imageNamePrefix": "hopper-",
"runOn": "main",
"runOn": "always",
"sccache": true,
"cudaComputeCap": 90,
"grpc": true,
"dockerfile": "Dockerfile-cuda"
},
{
"name": "All",
"imageNamePrefix": "cuda-",
"runOn": "main",
"runOn": "always",
"sccache": false,
"grpc": false,
"dockerfile": "Dockerfile-cuda-all"
},
{
"name": "cpu",
"imageNamePrefix": "cpu-",
"runOn": "main",
"runOn": "always",
"sccache": true,
"grpc": true,
"dockerfile": "Dockerfile"
}
]
Loading

0 comments on commit a28ad28

Please sign in to comment.