1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
|
# Copyright (C) 2021 - 2022 Advanced Micro Devices, Inc. All rights reserved.
#
# Permission is hereby granted, free of charge, to any person obtaining a copy
# of this software and associated documentation files (the "Software"), to deal
# in the Software without restriction, including without limitation the rights
# to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
# copies of the Software, and to permit persons to whom the Software is
# furnished to do so, subject to the following conditions:
#
# The above copyright notice and this permission notice shall be included in
# all copies or substantial portions of the Software.
#
# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
# THE SOFTWARE.
"""Utilities to build rocFFT."""
import subprocess
import logging
from . import git
from pathlib import Path
from .utils import sjoin
from shutil import which
def local(cmd, echo=True, **kwargs):
"""Run `cmd` using the shell.
Keyword arguments are passed down to `subprocess.run`.
"""
if echo:
print('local: ' + cmd)
logging.info('local: ' + cmd)
return subprocess.run(cmd, shell=True, **kwargs)
def local_amdgpu_target():
try:
for line in subprocess.Popen(
args=["rocminfo"], stdout=subprocess.PIPE).stdout.readlines():
if b'amdgcn-amd-amdhsa--' in line:
return line.split(b'--')[1].strip().decode('utf-8')
except:
pass
return ''
def build_rocfft(
commit,
dest=None,
repo='git@github.com:ROCmSoftwarePlatform/rocFFT-internal.git'):
"""Build public rocFFT (at specified git `commit`) and install into `dest`."""
top = Path('.').resolve() / ('rocFFT-' + commit)
if not top.exists():
git.clone(repo, top)
git.checkout(top, commit)
if git.is_dirty(top):
print(f'working directory {top} is dirty!')
raise SystemExit
build = top / 'build'
build.mkdir(exist_ok=True)
defs = [
'-DCMAKE_CXX_COMPILER=hipcc', '-DBUILD_CLIENTS_BENCH=ON',
'-DROCFFT_CALLBACKS_ENABLED=OFF', '-DSINGLELIB=ON',
'-DAMDGPU_TARGETS=' + local_amdgpu_target()
]
if dest:
defs += [f'-DCMAKE_INSTALL_PREFIX={dest}']
if which('ccache') is not None:
defs += ['-DCMAKE_CXX_COMPILER_LAUNCHER=ccache']
use_ninja = which('ninja') is not None
if use_ninja:
defs += ['-G', 'Ninja']
local(f'cmake {sjoin(defs)} ..', cwd=build, check=True)
if use_ninja:
local('ninja', cwd=build, check=True)
else:
local('make -j $(nproc)', cwd=build, check=True)
if dest:
if use_ninja:
local('ninja install', cwd=build, check=True)
else:
local('make install', cwd=build, check=True)
local(f'cp {build}/clients/staging/* {dest}')
|
