Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

feat: cuda miner support #167

Closed
wants to merge 26 commits into from
Closed
Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
26 commits
Select commit Hold shift + click to select a range
1b01ed8
feat: support cuda worker
Jun 18, 2024
16fccf3
opt: fallback to use gpu if no gpu
hunjixin Jun 18, 2024
f7de92d
feat: make worker number configurable
hunjixin Jun 18, 2024
89f8657
opt: make cuda run stablely
hunjixin Jun 18, 2024
c26b938
feat: little opt cuda
hunjixin Jun 19, 2024
46bd465
remove debug code
hunjixin Jun 19, 2024
05292be
feat: add flag
hunjixin Jun 20, 2024
992be81
fix: linux build
hunjixin Jun 20, 2024
04c3a30
feat: add flag to configura grid/block size
hunjixin Jun 20, 2024
9bfbe9d
fix: use cuda config
hunjixin Jun 20, 2024
e47263b
opt: reduce registre, 1/3 performance increase
hunjixin Jun 20, 2024
f0721f3
add a linux amd64 build job
richardbremner Jun 21, 2024
8b4d48b
Create linux_arm64.yml
richardbremner Jun 21, 2024
35e8cd9
run on push to branch
richardbremner Jun 21, 2024
e5656a0
Merge branch 'feat/cuda_binary' of github.com:Lilypad-Tech/lilypad in…
richardbremner Jun 21, 2024
c1a344b
use newer go version
richardbremner Jun 21, 2024
3c339c1
rename the github action
richardbremner Jun 21, 2024
eb1f510
add darwin arm64 build
richardbremner Jun 21, 2024
d586415
add linux arm64 build
richardbremner Jun 21, 2024
4dad55f
fix typo in workflow name
richardbremner Jun 21, 2024
af85961
fix some typos
richardbremner Jun 21, 2024
4c57b0b
add darwin amd64 build
richardbremner Jun 21, 2024
ed635fa
upload built binary artifacts
richardbremner Jun 21, 2024
dd560c9
feat: support cuda worker
Jun 18, 2024
9375723
github action to perform github binary releases
richardbremner Jun 21, 2024
a01973b
Merge branch 'feat/cuda_binary' into feat/cuda_miner_support
richardbremner Jun 21, 2024
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
52 changes: 52 additions & 0 deletions .github/workflows/darwin_amd64.yml
Original file line number Diff line number Diff line change
@@ -0,0 +1,52 @@
name: Build darwin-amd64 binary (no gpu)

on:
push:
branches:
- feat/cuda_binary

jobs:
build:
runs-on: macos-13 #uses amd64

steps:
- uses: actions/checkout@v4

- name: Set up Go
uses: actions/setup-go@v4
with:
go-version: '1.22'

- name: Check architecture
run: uname -m

- name: Build for darwin/amd64
env:
GOOS: darwin
GOARCH: amd64
CGO_ENABLED: 1
run: |
echo "-------------- OS: ${GOOS} : Arch: ${GOARCH} ---------- start"
output_dir="build/${GOOS}-${GOARCH}"
output_name="lilypad-${GOOS}-${GOARCH}"
mkdir -p "${output_dir}"

# Debug: Print environment variables for the build
echo "Building for ${GOOS}/${GOARCH} with GOOS=$GOOS, GOARCH=$GOARCH"

echo "excluding CUDA. specify 'cuda' build tag to include it."

go build -o "${output_dir}/${output_name}" -v -ldflags="-X 'github.com/lilypad-tech/lilypad/cmd/lilypad.VERSION=$(git describe --tags --abbrev=0)' -X 'github.com/lilypad-tech/lilypad/cmd/lilypad.COMMIT_SHA=$(git rev-parse HEAD)'"

echo "-------------- OS: ${GOOS} : Arch: ${GOARCH} ---------- done"

- name: Upload binary
uses: actions/upload-artifact@v3
with:
name: darwin-amd64-binary
path: build/darwin-amd64/lilypad-darwin-amd64

- name: Verify Build Output
run: |
ls -la build/darwin-amd64
echo "Build complete"
52 changes: 52 additions & 0 deletions .github/workflows/darwin_arm64.yml
Original file line number Diff line number Diff line change
@@ -0,0 +1,52 @@
name: Build darwin-arm64 binary (no gpu)

on:
push:
branches:
- feat/cuda_binary

jobs:
build:
runs-on: macos-latest #uses M1

steps:
- uses: actions/checkout@v4

- name: Set up Go
uses: actions/setup-go@v4
with:
go-version: '1.22'

- name: Check architecture
run: uname -m

- name: Build for darwin/arm64
env:
GOOS: darwin
GOARCH: arm64
CGO_ENABLED: 1
run: |
echo "-------------- OS: ${GOOS} : Arch: ${GOARCH} ---------- start"
output_dir="build/${GOOS}-${GOARCH}"
output_name="lilypad-${GOOS}-${GOARCH}"
mkdir -p "${output_dir}"

# Debug: Print environment variables for the build
echo "Building for ${GOOS}/${GOARCH} with GOOS=$GOOS, GOARCH=$GOARCH"

echo "excluding CUDA. specify 'cuda' build tag to include it."

go build -o "${output_dir}/${output_name}" -v -ldflags="-X 'github.com/lilypad-tech/lilypad/cmd/lilypad.VERSION=$(git describe --tags --abbrev=0)' -X 'github.com/lilypad-tech/lilypad/cmd/lilypad.COMMIT_SHA=$(git rev-parse HEAD)'"

echo "-------------- OS: ${GOOS} : Arch: ${GOARCH} ---------- done"

- name: Upload binary
uses: actions/upload-artifact@v3
with:
name: darwin-arm64-binary
path: build/darwin-arm64/lilypad-darwin-arm64

- name: Verify Build Output
run: |
ls -la build/darwin-arm64
echo "Build complete"
66 changes: 66 additions & 0 deletions .github/workflows/linux_amd64.yml
Original file line number Diff line number Diff line change
@@ -0,0 +1,66 @@
name: Build linux-amd64 binary (GPU)

on:
push:
branches:
- feat/cuda_binary

jobs:

build:
runs-on: ubuntu-latest
steps:
- uses: actions/checkout@v4

- name: Set up Go
uses: actions/setup-go@v4
with:
go-version: '1.22'

- name: Add NVIDIA package repository
run: |
sudo apt-get install -y gnupg2 curl
sudo mkdir -p /usr/share/keyrings
curl -fsSL https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2404/x86_64/3bf863cc.pub | sudo gpg --dearmor -o /usr/share/keyrings/nvidia-drivers.gpg
echo "deb [signed-by=/usr/share/keyrings/nvidia-drivers.gpg] https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2404/x86_64/ /" | sudo tee /etc/apt/sources.list.d/nvidia-drivers.list

- name: Update package lists
run: sudo apt-get update || true # Ignore errors from this command

- name: Install NVIDIA CUDA Toolkit
run: |
sudo apt-get install -y nvidia-cuda-toolkit
export PATH=/usr/local/cuda/bin:$PATH

- name: Build for linux/amd64
env:
GOOS: linux
GOARCH: amd64
CGO_ENABLED: 1
CUDA_HOME: /usr/local/cuda
LD_LIBRARY_PATH: /usr/local/cuda/lib64:$LD_LIBRARY_PATH
run: |
echo "-------------- OS: ${GOOS} : Arch: ${GOARCH} ---------- start"
output_dir="build/${GOOS}-${GOARCH}"
output_name="lilypad-${GOOS}-${GOARCH}"
mkdir -p "${output_dir}"

# Debug: Print environment variables for the build
echo "Building for ${GOOS}/${GOARCH} with GOOS=$GOOS, GOARCH=$GOARCH"
echo "CUDA_HOME: $CUDA_HOME"
echo "LD_LIBRARY_PATH: $LD_LIBRARY_PATH"

go build -o "${output_dir}/${output_name}" -v -tags cuda -ldflags="-X 'github.com/lilypad-tech/lilypad/cmd/lilypad.VERSION=$(git describe --tags --abbrev=0)' -X 'github.com/lilypad-tech/lilypad/cmd/lilypad.COMMIT_SHA=$(git rev-parse HEAD)'"

echo "-------------- OS: ${GOOS} : Arch: ${GOARCH} ---------- done"

- name: Upload binary
uses: actions/upload-artifact@v3
with:
name: linux-amd64-binary
path: build/linux-amd64/lilypad-linux-amd64

- name: Verify Build Output
run: |
ls -la build/linux-amd64
echo "Build complete"
66 changes: 66 additions & 0 deletions .github/workflows/linux_arm64.yml
Original file line number Diff line number Diff line change
@@ -0,0 +1,66 @@
name: Build linux-arm64 binary (GPU)

on:
push:
branches:
- feat/cuda_binary

jobs:

build:
runs-on: linux-arm64
steps:
- uses: actions/checkout@v4

- name: Set up Go
uses: actions/setup-go@v4
with:
go-version: '1.22'

- name: Add NVIDIA package repository
run: |
sudo apt-get install -y gnupg2 curl
sudo mkdir -p /usr/share/keyrings
curl -fsSL https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2404/x86_64/3bf863cc.pub | sudo gpg --dearmor -o /usr/share/keyrings/nvidia-drivers.gpg
echo "deb [signed-by=/usr/share/keyrings/nvidia-drivers.gpg] https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2404/x86_64/ /" | sudo tee /etc/apt/sources.list.d/nvidia-drivers.list

- name: Update package lists
run: sudo apt-get update || true # Ignore errors from this command

- name: Install NVIDIA CUDA Toolkit
run: |
sudo apt-get install -y nvidia-cuda-toolkit
export PATH=/usr/local/cuda/bin:$PATH

- name: Build for linux/arm64
env:
GOOS: linux
GOARCH: arm64
CGO_ENABLED: 1
CUDA_HOME: /usr/local/cuda
LD_LIBRARY_PATH: /usr/local/cuda/lib64:$LD_LIBRARY_PATH
run: |
echo "-------------- OS: ${GOOS} : Arch: ${GOARCH} ---------- start"
output_dir="build/${GOOS}-${GOARCH}"
output_name="lilypad-${GOOS}-${GOARCH}"
mkdir -p "${output_dir}"

# Debug: Print environment variables for the build
echo "Building for ${GOOS}/${GOARCH} with GOOS=$GOOS, GOARCH=$GOARCH"
echo "CUDA_HOME: $CUDA_HOME"
echo "LD_LIBRARY_PATH: $LD_LIBRARY_PATH"

go build -o "${output_dir}/${output_name}" -v -tags cuda -ldflags="-X 'github.com/lilypad-tech/lilypad/cmd/lilypad.VERSION=$(git describe --tags --abbrev=0)' -X 'github.com/lilypad-tech/lilypad/cmd/lilypad.COMMIT_SHA=$(git rev-parse HEAD)'"

echo "-------------- OS: ${GOOS} : Arch: ${GOARCH} ---------- done"

- name: Upload binary
uses: actions/upload-artifact@v3
with:
name: linux-arm64-binary
path: build/linux-arm64/lilypad-linux-arm64

- name: Verify Build Output
run: |
ls -la build/linux-arm64
echo "Build complete"
101 changes: 101 additions & 0 deletions .github/workflows/release_binaries.yml
Original file line number Diff line number Diff line change
@@ -0,0 +1,101 @@
name: Release binaries

on:
workflow_run:
workflows:
- linux_amd64.yml
- linux_arm64.yml
- darwin_amd64.yml
- darwin_arm64.yml
types:
- completed

jobs:
release:
if: github.ref == 'refs/heads/main' # Only run this job on the main branch
runs-on: ubuntu-latest
steps:
- uses: actions/checkout@v4

- name: Set up Git
run: |
git fetch --tags
SHORTSHA=$(git rev-parse --short HEAD)
echo "SHORTSHA=${SHORTSHA}" >> $GITHUB_ENV

- name: Download linux-amd64 binary
uses: actions/download-artifact@v3
with:
name: linux-amd64-binary
path: /tmp/build/lilypad-linux-amd64

- name: Download linux-arm64 binary
uses: actions/download-artifact@v3
with:
name: linux-arm64-binary
path: /tmp/build/lilypad-linux-arm64

- name: Download darwin-amd64 binary
uses: actions/download-artifact@v3
with:
name: darwin-amd64-binary
path: /tmp/build/lilypad-darwin-amd64

- name: Download darwin-arm64 binary
uses: actions/download-artifact@v3
with:
name: darwin-arm64-binary
path: /tmp/build/lilypad-darwin-arm64

- name: Create Release
id: create_release
uses: actions/create-release@v1
env:
GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }}
with:
tag_name: v2.0.0-${{ env.SHORTSHA }}
release_name: "Release v2.0.0-${{ env.SHORTSHA }}"
body: "Release v2.0.0-${{ env.SHORTSHA }}"
draft: false
prerelease: false


- name: Upload Release Asset - linux-amd64
uses: actions/upload-release-asset@v1
env:
GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }}
with:
upload_url: ${{ steps.create_release.outputs.upload_url }}
asset_path: /tmp/build/lilypad-linux-amd64
asset_name: lilypad-linux-amd64
asset_content_type: application/octet-stream

- name: Upload Release Asset - linux-arm64
uses: actions/upload-release-asset@v1
env:
GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }}
with:
upload_url: ${{ steps.create_release.outputs.upload_url }}
asset_path: /tmp/build/lilypad-linux-arm64
asset_name: lilypad-linux-arm64
asset_content_type: application/octet-stream

- name: Upload Release Asset - darwin-arm64
uses: actions/upload-release-asset@v1
env:
GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }}
with:
upload_url: ${{ steps.create_release.outputs.upload_url }}
asset_path: /tmp/build/lilypad-darwin-arm64
asset_name: lilypad-darwin-arm64
asset_content_type: application/octet-stream

- name: Upload Release Asset - darwin-amd64
uses: actions/upload-release-asset@v1
env:
GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }}
with:
upload_url: ${{ steps.create_release.outputs.upload_url }}
asset_path: /tmp/build/lilypad-darwin-amd64
asset_name: lilypad-darwin-amd64
asset_content_type: application/octet-stream
4 changes: 4 additions & 0 deletions go.mod
Original file line number Diff line number Diff line change
Expand Up @@ -15,13 +15,17 @@ require (
github.com/hashicorp/golang-lru/v2 v2.0.7
github.com/holiman/uint256 v1.2.4
github.com/ipfs/go-merkledag v0.11.0
github.com/pkg/errors v0.9.1
github.com/rs/zerolog v1.31.0
github.com/spf13/cobra v1.7.0
github.com/stretchr/testify v1.8.4
github.com/theckman/yacspin v0.13.12
gorgonia.org/cu v0.9.6
k8s.io/apimachinery v0.28.3
)

replace gorgonia.org/cu => github.com/hunjixin/cu v0.0.0-20240618140529-d11ba74b75b6

require (
dario.cat/mergo v1.0.0 // indirect
github.com/Microsoft/go-winio v0.6.1 // indirect
Expand Down
Loading
Loading