Skip to content

Commit 54ce994

Browse files
authored
Avoid aggregate graph access descriptor initialization (#2688)
1 parent 64afb35 commit 54ce994

1 file changed

Lines changed: 5 additions & 5 deletions

File tree

cuda_core/cuda/core/graph/_graph_node.pyx

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -845,18 +845,18 @@ cdef inline AllocNode GN_alloc(GraphNode self, size_t size, object device,
845845
num_deps = 1
846846
847847
cdef vector[cydriver.CUmemAccessDesc] access_descs
848+
cdef cydriver.CUmemAccessDesc access_desc
848849
cdef int peer_id
849850
cdef list peer_ids = []
850851
851852
if peer_access is not None:
852853
for peer_dev in peer_access:
853854
peer_id = getattr(peer_dev, 'device_id', peer_dev)
854855
peer_ids.append(peer_id)
855-
access_descs.push_back(cydriver.CUmemAccessDesc_st(
856-
cumemlocation_from_id(
857-
cydriver.CUmemLocationType.CU_MEM_LOCATION_TYPE_DEVICE, peer_id),
858-
cydriver.CUmemAccess_flags.CU_MEM_ACCESS_FLAGS_PROT_READWRITE,
859-
))
856+
access_desc.location = cumemlocation_from_id(
857+
cydriver.CUmemLocationType.CU_MEM_LOCATION_TYPE_DEVICE, peer_id)
858+
access_desc.flags = cydriver.CUmemAccess_flags.CU_MEM_ACCESS_FLAGS_PROT_READWRITE
859+
access_descs.push_back(access_desc)
860860
861861
cdef str memory_type_str = "device" if memory_type is None else str(memory_type)
862862

0 commit comments

Comments
 (0)