in Benchmarks/NVIDIA/GEMMCublasLt.py [0:0]
def __init__(self, path: str, machine: str, b: int = 1, i: int = 1000, w: int = 10000):
self.name = "GEMMCublasLt"
config = self.get_config(path)
self.m, self.n, self.k, self.duration, self.datatype = self.config_conversion(config)
self.b = b
self.i = i
self.w = w
self.bindir = ''
self.machine_name = machine
self.buffer = []
# A100 does not support fp8
if "A100" in machine:
self.datatype = "fp16"