Coverage for cuda / core / system / _device_utils.pxi: 90.91%
11 statements
« prev ^ index » next coverage.py v7.14.0, created at 2026-05-22 01:37 +0000
1# SPDX-FileCopyrightText: Copyright (c) 2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
2#
3# SPDX-License-Identifier: Apache-2.0
5from libc.stdint cimport uint64_t
8cpdef inline list[int] _unpack_bitmask(uint64_t[:] arr):
9 """
10 Unpack a list of integers containing bitmasks.
11 """
12 cdef uint64_t i, j, idx
13 cdef int mask_bits = 64 1abcdefghi
15 res = [] 1abcdefghi
17 for i in range(len(arr)): 1abcdefghi
18 cpu_offset = i * mask_bits 1abcdefghi
19 idx = 1 1abcdefghi
20 for j in range(mask_bits): 1abcdefghi
21 if arr[i] & idx: 1abcdefghi
22 res.append(cpu_offset + j) 1abcdefghi
23 idx <<= 1 1abcdefghi
24 return res 1abcdefghi