Skip to content

Commit 50c45de

Browse files
committed
Fix queue_ref attribute reference
1 parent 96dea84 commit 50c45de

File tree

2 files changed

+9
-5
lines changed

2 files changed

+9
-5
lines changed

numba_dpex/dpnp_iface/_intrinsic.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -73,8 +73,7 @@ def make_queue(context, builder, py_dpctl_sycl_queue):
7373
pyapi, py_dpctl_sycl_queue_addr, queue_struct_voidptr
7474
)
7575

76-
queue_struct = builder.load(queue_struct_ptr)
77-
queue_ref = builder.extract_value(queue_struct, 1)
76+
queue_ref = queue_struct_proxy.queue_ref
7877

7978
return_values = namedtuple(
8079
"return_values", "queue_ref queue_address_ptr pyapi"

numba_dpex/tests/core/types/DpctlSyclQueue/test_queue_ref_attr.py

Lines changed: 8 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -31,16 +31,21 @@ def are_queues_equal(typingctx, ty_queue1, ty_queue2):
3131

3232
# defines the custom code generation
3333
def codegen(context, builder, sig, args):
34+
q1 = cgutils.create_struct_proxy(ty_queue1)(
35+
context, builder, value=args[0]
36+
)
37+
q2 = cgutils.create_struct_proxy(ty_queue2)(
38+
context, builder, value=args[1]
39+
)
40+
3441
fnty = llvmir.FunctionType(
3542
cgutils.bool_t, [cgutils.voidptr_t, cgutils.voidptr_t]
3643
)
3744
fn = cgutils.get_or_insert_function(
3845
builder.module, fnty, "DPCTLQueue_AreEq"
3946
)
40-
qref1 = builder.extract_value(args[0], 1)
41-
qref2 = builder.extract_value(args[1], 1)
4247

43-
ret = builder.call(fn, [qref1, qref2])
48+
ret = builder.call(fn, [q1.queue_ref, q2.queue_ref])
4449

4550
return ret
4651

0 commit comments

Comments
 (0)