Skip to content

Commit a70b3df

Browse files
authored
Simplify LLAMA gemm specs (#38)
Drop unused parameter `gcount`
1 parent 19c832f commit a70b3df

File tree

2 files changed

+594
-594
lines changed

2 files changed

+594
-594
lines changed

gemmbench/gemm_utils.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,3 @@
1-
from utils import *
21
from dataclasses import dataclass
32
from pathlib import Path
43
from typing import Optional
@@ -10,7 +9,8 @@
109
get_default_run_config,
1110
get_default_scheduling_params,
1211
)
13-
import torch
12+
from utils import *
13+
import os
1414
import traceback
1515

1616

0 commit comments

Comments
 (0)