in Benchmarks/NVIDIA/HBMBandwidth.py [0:0]
def build(self):
current = os.getcwd()
path = "BabelStream"
isdir = os.path.isdir(path)
if not isdir:
results = subprocess.run(
["git", "clone", "https://github.com/gitaumark/BabelStream", path],
stdout=subprocess.PIPE,
stderr=subprocess.PIPE,
)
tools.write_log(tools.check_error(results))
build_path = os.path.join(current, "BabelStream")
os.chdir(build_path)
babelstream_build_path = os.path.join(build_path, "build")
arch ="sm_90"
if "A100" in self.machine_name:
arch = "sm_80"
if not os.path.isdir(babelstream_build_path):
os.mkdir(babelstream_build_path)
os.chdir(babelstream_build_path)
results = subprocess.run(
[
"cmake",
"../",
"-DMODEL=cuda",
"-DCUDA_ARCH=" + arch,
"-DCMAKE_CUDA_COMPILER=/usr/local/cuda/bin/nvcc",
],
stdout=subprocess.PIPE,
stderr=subprocess.PIPE,
)
tools.write_log(tools.check_error(results))
results = subprocess.run(
["make"], stdout=subprocess.PIPE, stderr=subprocess.PIPE
)
tools.write_log(tools.check_error(results))
else:
os.chdir(babelstream_build_path)