def __init__(self, arch: str):
super().__init__('fill', arch, basic_repeat_times=10)
fill_container = Container()
fill_container.update({'sparse': None}) # None: implement by feature
self.create_plan(fill_container, DataType(), DataSize(), MetricType())
# use tag_list to label the customized implementation (funcs).
self.add_func(['field'], fill_default)
self.add_func(['ndarray'], fill_default)
self.add_func(['sparse'], fill_sparse)
def __init__(self, arch: str):
super().__init__('atomic_ops', arch, basic_repeat_times=10)
atomic_ops = AtomicOps()
atomic_ops.remove(
['atomic_sub', 'atomic_and', 'atomic_xor', 'atomic_max'])
self.create_plan(atomic_ops, Container(), DataType(), DataSize(),
MetricType())
self.add_func(['field'], reduction_default)
self.add_func(['ndarray'], reduction_default)
def __init__(self, arch: str):
super().__init__('stencil_2d', arch, basic_repeat_times=10)
container = Container()
container.update({'sparse': None}) # None: implement by feature
self.create_plan(Scatter(), BloclLocalStorage(), container, DataType(),
DataSize2D(), MetricType())
# no use for field & ndarray
self.remove_cases_with_tags(['field', 'bls1'])
self.remove_cases_with_tags(['ndarray', 'bls1'])
self.add_func(['field'], stencil_2d_default)
self.add_func(['ndarray'], stencil_2d_default)
self.add_func(['sparse'], stencil_2d_sparse_bls)
def __init__(self, arch: str):
super().__init__('memcpy', arch, basic_repeat_times=10)
self.create_plan(Container(), DataType(), DataSize(), MetricType())
self.add_func(['field'], memcpy_default)
self.add_func(['ndarray'], memcpy_default)