trafo: add FLashAttention-2 for fused SoftMax #535
Workflow file for this run
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| # Copyright 2020 ETH Zurich and University of Bologna. | |
| # Licensed under the Apache License, Version 2.0, see LICENSE for details. | |
| # SPDX-License-Identifier: Apache-2.0 | |
| # Run functional regression checks | |
| name: ci | |
| on: [push, pull_request] | |
| jobs: | |
| ######## | |
| # Docs # | |
| ######## | |
| docs: | |
| name: Build documentation | |
| runs-on: ubuntu-22.04 | |
| container: | |
| image: ghcr.io/pulp-platform/snitch_cluster:ipc-verification | |
| steps: | |
| - uses: actions/checkout@v2 | |
| - name: Build docs | |
| run: mkdocs build | |
| ############################################## | |
| # Simulate SW on Snitch Cluster w/ Verilator # | |
| ############################################## | |
| sw-snitch-cluster-vlt: | |
| name: Simulate SW on Snitch Cluster w/ Verilator | |
| runs-on: ubuntu-22.04 | |
| container: | |
| image: ghcr.io/pulp-platform/snitch_cluster:ipc-verification | |
| steps: | |
| - uses: actions/checkout@v2 | |
| with: | |
| submodules: 'recursive' | |
| - name: Build Software | |
| run: | | |
| bender vendor init | |
| git submodule update --init --recursive | |
| make -C target/snitch_cluster sw | |
| - name: Build Hardware | |
| run: | | |
| make -C target/snitch_cluster bin/snitch_cluster.vlt | |
| - name: Run Tests | |
| working-directory: target/snitch_cluster | |
| run: |- | |
| ./sw/tests/run.py sw/tests/run.yaml --simulator verilator | |
| ./sw/apps/run.py sw/apps/run.yaml --simulator verilator | |
| ############################################ | |
| # Build SW on Snitch Cluster w/ Banshee # | |
| ############################################ | |
| sw-snitch-cluster-banshee: | |
| name: Simulate SW on Snitch Cluster w/ Banshee | |
| runs-on: ubuntu-22.04 | |
| container: | |
| image: ghcr.io/pulp-platform/snitch_cluster:ipc-verification | |
| steps: | |
| - uses: actions/checkout@v2 | |
| with: | |
| submodules: 'recursive' | |
| - name: Build Software | |
| run: | | |
| bender vendor init | |
| git submodule update --init --recursive | |
| make -C target/snitch_cluster SELECT_RUNTIME=banshee sw | |
| - name: Run Tests | |
| env: | |
| SNITCH_LOG: info | |
| working-directory: target/snitch_cluster | |
| run: |- | |
| rustup install 1.63.0 | |
| rustup override set 1.63.0 | |
| ./sw/tests/run.py sw/tests/run.yaml --simulator banshee | |
| ./sw/apps/run.py sw/apps/run.yaml --simulator banshee |