gpu ci 3

mspronesti · Jul 29, 2024 · 41f0b51 · 41f0b51
1 parent 3c169bd
commit 41f0b51
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 6 deletions.
diff --git a/.github/workflows/ci-dev-gpu.yml b/.github/workflows/ci-dev-gpu.yml
@@ -1,4 +1,4 @@
-name: CI kernels (CPU)
+name: CI kernels (GPU)
 
 on: [push, pull_request]
 
@@ -11,14 +11,10 @@ jobs:
 
       - name: Build kernels
         run:  |
-          qsub -I -l nodes=1:gpu:ppn=2 -d .
-
           cd dev/sycl
           make CC=icx
 
       - name: Run kernels
         run: |
-          qsub -I -l nodes=1:gpu:ppn=2 -d .
-
           cd dev/sycl
           make run_all CC=icx
diff --git a/dev/sycl/Makefile b/dev/sycl/Makefile
@@ -27,7 +27,7 @@ ifeq ($(HIP),yes)
     CFLAGS += -fsycl-targets=amdgcn-amd-amdhsa -Xsycl-target-backend --offload-arch=$(HIP_ARCH)
 endif
 
-TARGETS = adamw attention_backward attention_forward classifier_fused crossentropy_forward crossentropy_softmax_backward encoder_backward encoder_forward gelu_backward gelu_forward global_norm layernorm_backward layernorm_forward matmul_backward matmul_backward_bias matmul_forward residual_forward  trimat_forward fused_residual_forward
+TARGETS = $(patsubst %.cpp,%,$(wildcard *.cpp))
 
 all: $(TARGETS)
 

diff --git a/dev/sycl/adamw.cpp b/dev/sycl/adamw.cpp
@@ -128,6 +128,8 @@ int main(int argc, char** argv) {
 
     // Allocate device memory
     sycl::queue q(sycl::default_selector_v, sycl::property::queue::in_order());
+    std::cout << "Running on: " << q.get_device().get_info<sycl::info::device::name>() << "\n
+
     float* d_params = sycl::malloc_device<float>(num_parameters, q);
     float* d_grads = sycl::malloc_device<float>(num_parameters, q);
     float* d_m_memory = sycl::malloc_device<float>(num_parameters, q);