From 57e3327342344508381c8747aaefe5cc5a5cae84 Mon Sep 17 00:00:00 2001 From: Derek Christ Date: Thu, 7 Mar 2024 16:46:59 +0100 Subject: [PATCH] Cleanup --- benches.py | 2 -- run_microbenchmarks.py | 2 +- workloads.py | 20 -------------------- 3 files changed, 1 insertion(+), 23 deletions(-) delete mode 100644 workloads.py diff --git a/benches.py b/benches.py index 2f4badf..b7b3e57 100644 --- a/benches.py +++ b/benches.py @@ -6,8 +6,6 @@ import json import polars as pl import dataclasses -import workloads as wl - from config import Statistics, Configuration device = torch.device("cuda:0") diff --git a/run_microbenchmarks.py b/run_microbenchmarks.py index 4b42b17..1cb45fb 100644 --- a/run_microbenchmarks.py +++ b/run_microbenchmarks.py @@ -33,4 +33,4 @@ for workload in workloads: print(result) df = pl.DataFrame(results) -df.write_csv("rocm_results.csv") +df.write_csv("vega_results.csv") diff --git a/workloads.py b/workloads.py deleted file mode 100644 index 76f5db8..0000000 --- a/workloads.py +++ /dev/null @@ -1,20 +0,0 @@ -import torch - -def gemv(matrix, input_vector): - return torch.matmul(matrix, input_vector) - -def gemv_layers(matrix, input_vector): - for _ in range(5): - input_vector = torch.matmul(matrix, input_vector) - input_vector.relu() - - return input_vector - -def vadd(vector_a, vector_b): - return torch.add(vector_a, vector_b) - -def vmul(vector_a, vector_b): - return torch.mul(vector_a, vector_b) - -def haxpy(vector_a, vector_b): - return torch.add(vector_a, vector_b, alpha=2)