def get_build_kwargs(self):
kwargs = {}
- if 'cuda' in self.task.target.keys or 'opencl' in self.task.target.keys:
+ if 'cuda' in self.task.target.keys or 'opencl' in self.task.target.keys or \
+ 'rocm' in self.task.target.keys:
remote = request_remote(self.key, self.host, self.port)
ctx = remote.context(str(self.task.target), 0)
max_dims = ctx.max_thread_dimensions
def _build_func_common(measure_input, check_gpu=None, cuda_arch=None, build_option=None):
"""Common part for building a configuration"""
target, task, config = measure_input
-
with target:
s, args = task.instantiate(config)