Simplify LLAMA gemm specs (#38)

Drop unused parameter `gcount`
nod-ai · Jan 6, 2025 · a70b3df · a70b3df
1 parent 19c832f
commit a70b3df
Show file tree

Hide file tree

Showing 2 changed files with 594 additions and 594 deletions.
diff --git a/gemmbench/gemm_utils.py b/gemmbench/gemm_utils.py
@@ -1,4 +1,3 @@
-from utils import *
 from dataclasses import dataclass
 from pathlib import Path
 from typing import Optional
@@ -10,7 +9,8 @@
     get_default_run_config,
     get_default_scheduling_params,
 )
-import torch
+from utils import *
+import os
 import traceback