Coverage for cuda/core/system/_device_utils.pxi: 90.91%

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

2# 

3# SPDX-License-Identifier: Apache-2.0 

4  

5from libc.stdint cimport uint64_t 

6  

7  

8def _unpack_bitmask(arr: object) -> list[int]: 

9 """ 

10 Unpack a list of integers containing bitmasks. 

11 """ 

12 cdef uint64_t i, j, idx 

13 cdef int mask_bits = 64 10 ctx1abcdefghij

14  

15 res = [] 10 ctx1abcdefghij

16  

17 for i in range(len(arr)): 10 ctx1abcdefghij

18 cpu_offset = i * mask_bits 9 ctx1abcdefghi

19 idx = 1 9 ctx1abcdefghi

20 for j in range(mask_bits): 9 ctx1abcdefghi

21 if arr[i] & idx: 9 ctx1abcdefghi

22 res.append(cpu_offset + j) 9 ctx1abcdefghi

23 idx <<= 1 9 ctx1abcdefghi

24 return res 9 ctx1abcdefghi