File: GB_clear_matrix_header.h

package info (click to toggle)
suitesparse 1%3A7.10.1%2Bdfsg-1
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid, trixie
  • size: 254,920 kB
  • sloc: ansic: 1,134,743; cpp: 46,133; makefile: 4,875; fortran: 2,087; java: 1,826; sh: 996; ruby: 725; python: 495; asm: 371; sed: 166; awk: 44
file content (68 lines) | stat: -rw-r--r-- 2,650 bytes parent folder | download | duplicates (2)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
//------------------------------------------------------------------------------
// GB_clear_matrix_header.h: macros for allocating static headers
//------------------------------------------------------------------------------

// SuiteSparse:GraphBLAS, Timothy A. Davis, (c) 2017-2025, All Rights Reserved.
// SPDX-License-Identifier: Apache-2.0

//------------------------------------------------------------------------------

// By default, many internal temporary matrices use statically allocated
// headers to reduce the number of calls to malloc/free.  This works fine for
// matrices on the CPU, but the static headers do not get automatically
// transfered to the GPU.  Only dynamically allocated headers, allocated by
// rmm_wrap_malloc, get transfered.  Set this to 1 to turn off static headers
// (required for CUDA).  Leave static headers enabled by default by leaving
// this commented out or setting GBNSTATIC to 0.

#ifndef GBNSTATIC
    #if defined ( GRAPHBLAS_HAS_CUDA )
    #define GBNSTATIC 1
    #else
    #define GBNSTATIC 0
    #endif
#endif

#undef GB_CLEAR_MATRIX_HEADER

#if GBNSTATIC

    // do not use any static headers
    #define GB_CLEAR_MATRIX_HEADER(XX,XX_header_handle)                     \
    {                                                                       \
        size_t XX_size ;                                                    \
        XX = GB_CALLOC_MEMORY (1, sizeof (struct GB_Matrix_opaque),         \
            &XX_size) ;                                                     \
        if (XX == NULL)                                                     \
        {                                                                   \
            GB_FREE_ALL ;                                                   \
            return (GrB_OUT_OF_MEMORY) ;                                    \
        }                                                                   \
        XX->header_size = XX_size ;                                         \
        XX->magic = GB_MAGIC2 ;                                             \
    }

#else

    // use static headers
    #define GB_CLEAR_MATRIX_HEADER(XX,XX_header_handle)                     \
    {                                                                       \
        XX = GB_clear_matrix_header (XX_header_handle) ;                    \
    }

#endif

#ifndef GB_CLEAR_MATRIX_HEADER_H
#define GB_CLEAR_MATRIX_HEADER_H

static inline GrB_Matrix GB_clear_matrix_header // clear a static header
(
    GrB_Matrix C    // static header to clear
)
{
    memset (C, 0, sizeof (struct GB_Matrix_opaque)) ;
    return (C) ;
}

#endif