Skip to content

Commit 52e6937

Browse files
committed
Avoid aggregate graph access descriptor initialization
1 parent 1ca15a3 commit 52e6937

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
@@ -828,18 +828,18 @@ cdef inline AllocNode GN_alloc(GraphNode self, size_t size, object device,
828828
num_deps = 1
829829
830830
cdef vector[cydriver.CUmemAccessDesc] access_descs
831+
cdef cydriver.CUmemAccessDesc access_desc
831832
cdef int peer_id
832833
cdef list peer_ids = []
833834
834835
if peer_access is not None:
835836
for peer_dev in peer_access:
836837
peer_id = getattr(peer_dev, 'device_id', peer_dev)
837838
peer_ids.append(peer_id)
838-
access_descs.push_back(cydriver.CUmemAccessDesc_st(
839-
cumemlocation_from_id(
840-
cydriver.CUmemLocationType.CU_MEM_LOCATION_TYPE_DEVICE, peer_id),
841-
cydriver.CUmemAccess_flags.CU_MEM_ACCESS_FLAGS_PROT_READWRITE,
842-
))
839+
access_desc.location = cumemlocation_from_id(
840+
cydriver.CUmemLocationType.CU_MEM_LOCATION_TYPE_DEVICE, peer_id)
841+
access_desc.flags = cydriver.CUmemAccess_flags.CU_MEM_ACCESS_FLAGS_PROT_READWRITE
842+
access_descs.push_back(access_desc)
843843
844844
cdef str memory_type_str = "device" if memory_type is None else str(memory_type)
845845

0 commit comments

Comments
 (0)