This project is designed to run SDPA tests using configurations specified in JSON format. It utilizes the submitit library to manage job submissions and generate reports in Markdown format.
| GPU | Report |
|---|---|
| V100 | report |
| H100 | report |
| H200 | report |
- Clone the repository:
git clone git@github.com:CAIIVS/sdpa-gpu-test.git cd sdpa-gpu-test.git
-
Prepare a JSON configuration file specifying the test parameters.
-
Run the main script with the configuration file:
sbatch runner.sh configs/v100.json
-
The results will be saved in the
outputsdirectory as a Markdown report.
The JSON configuration file should include the following fields:
gpu: The type of GPU to use.input_dims: List of input dimensions.input_channels: List of input channels.input_lengths: List of input lengths.backend_configs: List of backend configurations.output_path: (Optional) Directory to save the output report.timeout: (Optional) Timeout for each job in minutes.account: (Optional) SLURM account to use.partition: (Optional) SLURM partition to use.