move handling back to top

Author: brandon-b-miller

numba_cuda/numba/cuda/dispatcher.py

@@ -544,6 +544,8 @@ def _prepare_args(self, ty, val, stream, retr, kernelargs):
                 ty, val = extension.prepare_args(
                     ty, val, stream=stream, retr=retr
                 )
+        elif handler := _arg_handlers.get(type(val)):
+            ty, val = handler.prepare_args(ty, val, stream=stream, retr=retr)
 
         if isinstance(ty, types.Array):
             devary = wrap_arg(val).to_device(retr, stream)
@@ -620,8 +622,6 @@ def _prepare_args(self, ty, val, stream, retr, kernelargs):
                 )
             except NotImplementedError:
                 raise NotImplementedError(ty, val)
-        elif handler := _arg_handlers.get(type(val)):
-            ty, val = handler.prepare_args(ty, val, stream=stream, retr=retr)
         else:
             raise NotImplementedError(ty, val)