From e71ac17e1a1f97eb2826818e241b27924dad02ee Mon Sep 17 00:00:00 2001 From: Huamin Chen Date: Mon, 20 Nov 2023 09:45:16 -0500 Subject: [PATCH] update CI Signed-off-by: Huamin Chen --- .github/workflows/integration_test.yml | 3 ++- .github/workflows/integration_test_libbpf.yml | 3 ++- .github/workflows/unit_test.yml | 3 ++- pkg/model/estimator/local/xgboost_model_weight.go | 2 ++ 4 files changed, 8 insertions(+), 3 deletions(-) diff --git a/.github/workflows/integration_test.yml b/.github/workflows/integration_test.yml index b478556822..936e801ff3 100644 --- a/.github/workflows/integration_test.yml +++ b/.github/workflows/integration_test.yml @@ -22,9 +22,10 @@ jobs: go-version-file: go.mod # set up ebpf - name: install libbpf - uses: sustainable-computing-io/kepler-action@v0.0.2 + uses: sustainable-computing-io/kepler-action@v0.0.3 with: ebpfprovider: bcc + xgboost_version: 2.0.1 # build kepler image - name: build and export Kepler image run: | diff --git a/.github/workflows/integration_test_libbpf.yml b/.github/workflows/integration_test_libbpf.yml index 95639d2d63..da5c5b833f 100644 --- a/.github/workflows/integration_test_libbpf.yml +++ b/.github/workflows/integration_test_libbpf.yml @@ -22,9 +22,10 @@ jobs: go-version-file: go.mod # set up ebpf - name: install libbpf - uses: sustainable-computing-io/kepler-action@v0.0.2 + uses: sustainable-computing-io/kepler-action@v0.0.3 with: ebpfprovider: libbpf + xgboost_version: 2.0.1 # build kepler image - name: build and export Kepler image run: | diff --git a/.github/workflows/unit_test.yml b/.github/workflows/unit_test.yml index 08db56d7e0..be8ae50c0f 100644 --- a/.github/workflows/unit_test.yml +++ b/.github/workflows/unit_test.yml @@ -28,9 +28,10 @@ jobs: GOPATH: /home/runner/go GOBIN: /home/runner/go/bin - name: install bcc - uses: sustainable-computing-io/kepler-action@v0.0.2 + uses: sustainable-computing-io/kepler-action@v0.0.3 with: ebpfprovider: bcc + xgboost_version: 2.0.1 - name: Prepare environment run: | sudo apt-get install -y cpuid diff --git a/pkg/model/estimator/local/xgboost_model_weight.go b/pkg/model/estimator/local/xgboost_model_weight.go index f183b8bf58..a516f0371d 100644 --- a/pkg/model/estimator/local/xgboost_model_weight.go +++ b/pkg/model/estimator/local/xgboost_model_weight.go @@ -17,6 +17,7 @@ limitations under the License. package local /* +#cgo CFLAGS: -I/usr/local/include #cgo LDFLAGS: -lxgboost -L/usr/local/lib64 #include #include @@ -63,6 +64,7 @@ int Predict(BoosterHandle h_booster, float *data, int rows, int cols, float *out } */ import "C" + import ( "encoding/base64" "fmt"