diff --git a/python/dune/perftool/loopy/vcl.py b/python/dune/perftool/loopy/vcl.py index fb7c51f85db261d39f085ad2e3b08dd48b24c14a..9d044939ac3b7bf45ecff4d78e70ebe2dcdea9ca 100644 --- a/python/dune/perftool/loopy/vcl.py +++ b/python/dune/perftool/loopy/vcl.py @@ -102,3 +102,8 @@ def vcl_function_mangler(knl, func, arg_dtypes): vcl = lp.types.NumpyType(get_vcl_type(dtype)) return lp.CallMangleInfo(func, (vcl,), (vcl,)) + if isinstance(func, str) and func.endswith('.load'): + return lp.CallMangleInfo(func, (), (lp.types.NumpyType(np.int32),)) + + if isinstance(func, str) and func.endswith('.store'): + return lp.CallMangleInfo(func, (), (lp.types.NumpyType(np.int32),))