Coverage for cuda/core/_memory/_copy_attributes.pyx: 100.00%

1# SPDX-FileCopyrightText: Copyright (c) 2026 NVIDIA CORPORATION & AFFILIATES. All rights reserved. (empty)

2# 

3# SPDX-License-Identifier: Apache-2.0 

4  

5from libc.string cimport memset 

6  

7from cuda.bindings cimport cydriver 

8from cuda.core._memory._location cimport to_cumemlocation 

9  

10from cuda.core._memory._managed_location import _coerce_location (empty)

11  

12  

13cdef cydriver.CUmemcpyAttributes _to_cu_memcpy_attributes(object attr): (empty)

14 """Convert a CopyOptions to a cydriver.CUmemcpyAttributes struct.""" 

15 cdef cydriver.CUmemcpyAttributes cu_attr 

16 memset(&cu_attr, 0, sizeof(cydriver.CUmemcpyAttributes)) 35 ctx1IghijklmnopqrcdastuvwxyzefbABCDEFGH

17 cu_attr.srcAccessOrder = <cydriver.CUmemcpySrcAccessOrder>(<int>attr._to_driver_enum()) 34 ctx1ghijklmnopqrcdastuvwxyzefbABCDEFGH

18 cu_attr.flags = <unsigned int>(<int>attr._to_driver_flags()) 34 ctx1ghijklmnopqrcdastuvwxyzefbABCDEFGH

19  

20 cdef object src_loc = _coerce_location(attr.src_location_hint, allow_none=True) 34 ctx1ghijklmnopqrcdastuvwxyzefbABCDEFGH

21 cdef object dst_loc = _coerce_location(attr.dst_location_hint, allow_none=True) 35 ctx1IghijklmnopqrcdastuvwxyzefbABCDEFGH

22  

23 if src_loc is not None: 34 ctx1ghijklmnopqrcdastuvwxyzefbABCDEFGH

24 cu_attr.srcLocHint = to_cumemlocation(src_loc.kind, src_loc.id) 2 ctx1ab

25 if dst_loc is not None: 34 ctx1ghijklmnopqrcdastuvwxyzefbABCDEFGH

26 cu_attr.dstLocHint = to_cumemlocation(dst_loc.kind, dst_loc.id) 6 ctx1cdaefb

27  

28 return cu_attr 34 ctx1ghijklmnopqrcdastuvwxyzefbABCDEFGH