bug fixes

.github/workflows/run_bench.yml

@@ -1,10 +1,9 @@
 name: Build and Test
 
-on: 
-  push:
+on:
   pull_request:
   schedule:
-    - cron: '0 7 * * *'
+    - cron: '0 8 * * *'
 
 jobs:
   build:
@@ -58,9 +57,10 @@ jobs:
 
       - name: Run RocBLAS Benchmarks
         run: |
+          sudo pkill -f gemm-bench
           cd $GITHUB_WORKSPACE
           source $GITHUB_WORKSPACE/venv/bin/activate
-          sudo $GITHUB_WORKSPACE/build/gemm-bench --device=0 &
+          for device in $(seq 3 7); do (sudo $GITHUB_WORKSPACE/build/gemm-bench --device=$device &); done
           ./gb run --backends=rocblas --output=rocblas.hdf
           sudo pkill -f gemm-bench
           deactivate
@@ -69,7 +69,7 @@ jobs:
         run: |
           cd $GITHUB_WORKSPACE
           source $GITHUB_WORKSPACE/venv/bin/activate
-          sudo $GITHUB_WORKSPACE/build/gemm-bench --device=0 &
+          for device in $(seq 3 7); do (sudo $GITHUB_WORKSPACE/build/gemm-bench --device=$device &); done
           ./gb run --backends=iree --output=iree.hdf
           sudo pkill -f gemm-bench
           deactivate

.gitignore

@@ -0,0 +1,5 @@
+build
+rocm_gemm_venv
+*.venv
+__pycache__
+*.hdf

src/.gitignore

@@ -0,0 +1 @@
+ireekernelsbuild

src/benchmark/gemm-bench.hpp

@@ -81,13 +81,13 @@ namespace GEMMBench
     class GEMMPipeline
     {
     public:
-        void initialize() {};
-        void destroy() {};
-        void linkData(GEMMData* data)
+        virtual void initialize() {};
+        virtual void destroy() {};
+        virtual void linkData(GEMMData* data)
         {
             this->data = data;
         };
-        void           setDevice(int device_id) {};
+        virtual void           setDevice(int device_id) {};
         virtual Result run(Problem problem) = 0;
 
     protected:
@@ -104,17 +104,17 @@ namespace GEMMBench
 
     public:
         IREEGEMMBench(){};
-        void   initialize();
-        void   setDevice(int device_id);
-        void   destroy();
+        void   initialize() override;
+        void   setDevice(int device_id) override;
+        void   destroy() override;
         Result run(Problem problem) override;
     };
 
     class RocBLASGEMMBench : public GEMMPipeline
     {
     public:
         RocBLASGEMMBench(){};
-        void   setDevice(int device_id);
+        void   setDevice(int device_id) override;
         Result run(Problem problem) override;
     };
 

src/benchmark/run_iree.cpp

@@ -39,7 +39,6 @@ std::string
 
 void IREEGEMMBench::initialize()
 {
-    std::cout << "Initializing bench" << std::endl;
     char** compileArgs = new char*[2];
     compileArgs[0]     = (char*)"--iree-hal-target-backends=rocm";
     compileArgs[1]     = (char*)"--iree-rocm-target-chip=gfx942";