Skip to content

Commit c1af1e4

Browse files
committed
Add benchmarks for sgemm
1 parent af21aae commit c1af1e4

File tree

1 file changed

+26
-0
lines changed

1 file changed

+26
-0
lines changed
Lines changed: 26 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,26 @@
1+
from benchmarks.numpy.common import Benchmark
2+
from benchmarks.utils import sync
3+
from benchmarks.utils.helper import parameterize
4+
5+
from .utils import benchmark
6+
from .sgemm import sgemm
7+
8+
import cupy
9+
10+
11+
@sync
12+
@parameterize([('m', [1, 1500]),
13+
('n', [1, 1500]),
14+
('k', [1, 3000])])
15+
class SGEMM(Benchmark):
16+
def setup(self, m, n, k):
17+
low, high = -1, 1
18+
dtype = cupy.float32
19+
self.A = cupy.random.uniform(low, high, (m, k)).astype(dtype)
20+
self.B = cupy.random.uniform(low, high, (k, n)).astype(dtype)
21+
22+
def time_sgemm(self, m, n, k):
23+
sgemm(self.A, self.B)
24+
25+
def time_dot(self, m, n, k):
26+
cupy.dot(self.A, self.B)

0 commit comments

Comments
 (0)