diff --git a/loopy/transform/iname.py b/loopy/transform/iname.py index ed6fa46220259b664c2f1122c3628127fc3e4617..ae717e1c7550b697c23b9d032fd795c362e387f0 100644 --- a/loopy/transform/iname.py +++ b/loopy/transform/iname.py @@ -1240,7 +1240,9 @@ def remove_unused_inames(knl, inames=None): def remove_any_newly_unused_inames(transformation_func): + from functools import wraps + @wraps(transformation_func) def wrapper(knl, *args, **kwargs): # check for remove_unused_inames argument, default: True