Skip to content

Commit

Permalink
Generate different function sigs
Browse files Browse the repository at this point in the history
  • Loading branch information
khaled committed May 7, 2023
1 parent 4ce1f12 commit 6aa9782
Showing 1 changed file with 3 additions and 1 deletion.
Original file line number Diff line number Diff line change
Expand Up @@ -31,13 +31,15 @@ def {0:s}(x=None, queue=None):
@pytest.mark.parametrize("queue_arg", queue_args)
def test_dpnp_functions_basic(dpnp_function_param, queue_arg):
"""Basic queue handling in a basic dpnp function scenario"""
function_name = "func"

op_name = dpnp_function_param[0]
function_name = "func_{0:s}".format(op_name)
dpnpf_arg = ", ".join(str(v) for v in dpnp_function_param[1])
q_arg = (", " + queue_arg) if queue_arg != "" else ""
func_expr = func_template_basic.format(
function_name, op_name, dpnpf_arg + q_arg
)
print(func_expr)

func = compile_function(func_expr, function_name, globals())
f = dpjit(func)
Expand Down

0 comments on commit 6aa9782

Please sign in to comment.