oneDNN GH actions

Author: graemenail

.github/workflows/ubuntu.yml

@@ -107,6 +107,34 @@ jobs:
       working-directory: build
       run: make -j2
 
+    - name: Configure CMake (oneDNN)
+      if: matrix.cpu == true && matrix.gpu == false
+      id: cmake-onednn
+      run: |
+        mkdir -p build-onednn
+        cd build-onednn
+        CC=/usr/bin/gcc-${{ matrix.gcc }} CXX=/usr/bin/g++-${{ matrix.gcc }} CUDAHOSTCXX=/usr/bin/g++-${{ matrix.gcc }} \
+        cmake .. \
+          -DBoost_ARCHITECTURE=-x64 \
+          -DCMAKE_BUILD_TYPE=Debug \
+          -DCOMPILE_CPU=${{ matrix.cpu }} \
+          -DCOMPILE_CUDA=${{ matrix.gpu }} \
+          -DCOMPILE_EXAMPLES=${{ matrix.examples }} \
+          -DCOMPILE_SERVER=on \
+          -DCOMPILE_TESTS=${{ matrix.unit_tests }} \
+          -DCUDA_TOOLKIT_ROOT_DIR=/usr/local/cuda-${{ matrix.cuda }} \
+          -DDETERMINISTIC=on \
+          -DUSE_FBGEMM=${{ matrix.cpu }} \
+          -DUSE_SENTENCEPIECE=on \
+          -DUSE_STATIC_LIBS=on \
+          -DUSE_MKL=OFF -DUSE_DNNL=ON
+
+    - name: Compile (oneDNN)
+      if: steps.cmake-onednn.conclusion == 'success'
+      working-directory: build-onednn
+      run: make -j2 && make clean
+
+
     # TODO: add a flag to CMake to compile unit tests only on CPU
     - name: Run unit tests
       working-directory: build

.github/workflows/windows.yml

@@ -51,7 +51,7 @@ jobs:
         echo "$env:CUDA_PATH/bin"       | Out-File -FilePath $env:GITHUB_PATH -Encoding utf8 -Append
       shell: powershell
       if: matrix.gpu == true
-    # Cache boost install 
+    # Cache boost install
     - name: Cache Boost
       id: cache-boost
       uses: actions/cache@v3
@@ -109,6 +109,39 @@ jobs:
       working-directory: ${{ github.workspace }}/build/Debug
       run: cmake --build . --target clean
 
+    - name: Build Debug (oneDNN)
+      id: build-debug-onednn
+      uses: lukka/run-cmake@v3
+      with:
+        buildDirectory: ${{ github.workspace }}/build/Debug-oneDNN
+        cmakeAppendedArgs: '-G Ninja
+          -DCMAKE_BUILD_TYPE="Debug"
+          -DOPENSSL_USE_STATIC_LIBS="TRUE"
+          -DOPENSSL_MSVC_STATIC_RT="TRUE"
+          -DCOMPILE_CPU="TRUE"
+          -DCOMPILE_CUDA="${{ matrix.gpu }}"
+          -DCOMPILE_SERVER="FALSE"
+          -DCOMPILE_TESTS="TRUE"
+          -DDETERMINISTIC="TRUE"
+          -DUSE_FBGEMM="TRUE"
+          -DUSE_DNNL="TRUE" -DUSE_MKL="FALSE"
+          -DUSE_MPI="FALSE"
+          -DUSE_NCCL="FALSE"
+          -DUSE_SENTENCEPIECE="TRUE"
+          -DUSE_STATIC_LIBS="TRUE"'
+        cmakeListsOrSettingsJson: CMakeListsTxtAdvanced
+        cmakeListsTxtPath: ${{ github.workspace }}/CMakeLists.txt
+        useVcpkgToolchainFile: true
+      # Building in Debug is sufficient for the all-in CPU+GPU compilation;
+      # its main purpose is to detect warnings that the Release build is not
+      # able to find sometimes.
+      if: matrix.gpu == false
+
+    - name: Cleanup Debug
+      if: steps.build-debug-onednn.conclusion == 'success'
+      working-directory: ${{ github.workspace }}/build/Debug-oneDNN
+      run: cmake --build . --target clean
+
     # Windows CUDA builds use USE_NCCL=off due to compilation errors
     - name: Build Release
       uses: lukka/run-cmake@v3