[Intel] Fix test_function_arguments

Signed-off-by: Whitney Tsang <[email protected]>

Author: whitneywhtsang

python/test/unit/runtime/test_cache.py

@@ -577,7 +577,7 @@ def cache_hook(*args, **kwargs):
     assert pointer_range_32 == [(0, )]
 
 
-def test_function_arguments():
+def test_function_arguments(device):
 
     @triton.jit
     def func1():
@@ -601,7 +601,7 @@ def kernel(Y, fn: tl.constexpr, fn_args):
 
     JITFunction.cache_hook = None
     JITFunction.compiled_hook = None
-    y = torch.zeros((5, ), dtype=torch.int32, device="cuda")
+    y = torch.zeros((5, ), dtype=torch.int32, device=device)
     kernel[(1, )](y[0], func1, tuple())
     kernel[(1, )](y[1], func2, tuple())
     kernel[(1, )](y[2], func3, (3, ))