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
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
|
//------------------------------------------------------------------------------
// GraphBLAS/CUDA/GB_cuda.hpp: include file for host CUDA methods (not for JIT)
//------------------------------------------------------------------------------
// SuiteSparse:GraphBLAS, Timothy A. Davis, (c) 2017-2025, All Rights Reserved.
// This file: Copyright (c) 2024-2025, NVIDIA CORPORATION. All rights reserved.
// SPDX-License-Identifier: Apache-2.0
//------------------------------------------------------------------------------
#ifndef GB_CUDA_HPP
#define GB_CUDA_HPP
#include "CUDA/include/GraphBLAS_cuda.hpp"
extern "C"
{
#include <cassert>
#include <cmath>
#include "GB.h"
#include "jitifyer/GB_stringify.h"
}
// Finally, include the CUDA definitions
#include "cuda_runtime.h"
#include "cuda.h"
#include <limits>
#include <iostream>
#include <cstdint>
#include <stdint.h>
#include <stdio.h>
#include "CUDA/include/GB_cuda_error.hpp"
#include "CUDA/include/GB_cuda_timer.hpp"
//------------------------------------------------------------------------------
// prefetch and memadvise
//------------------------------------------------------------------------------
// for the "which" parameter of GB_cuda_matrix_prefetch:
// FIXME: rename this to GB_WHATEVER_P for GB_cuda_matrix_memadvise
#define GB_PREFETCH_P 1
#define GB_PREFETCH_H 2
#define GB_PREFETCH_Y 4
#define GB_PREFETCH_B 8
#define GB_PREFETCH_I 16
#define GB_PREFETCH_X 32
#define GB_PREFETCH_PIX (GB_PREFETCH_P + GB_PREFETCH_I + GB_PREFETCH_X)
#define GB_PREFETCH_PYI (GB_PREFETCH_P + GB_PREFETCH_Y + GB_PREFETCH_I)
#define GB_PREFETCH_PYBI (GB_PREFETCH_PYI + GB_PREFETCH_B)
#define GB_PREFETCH_PYBIX (GB_PREFETCH_PYBI + GB_PREFETCH_X)
#define GB_PREFETCH_PHI (GB_PREFETCH_P + GB_PREFETCH_H + GB_PREFETCH_I)
#define GB_PREFETCH_PHBI (GB_PREFETCH_PHI + GB_PREFETCH_B)
#define GB_PREFETCH_PHBIX (GB_PREFETCH_PHBI + GB_PREFETCH_X)
GrB_Info GB_cuda_matrix_prefetch
(
GrB_Matrix A,
int which, // which components to prefetch (phybix control)
int device, // GPU device or cudaCpuDeviceId
cudaStream_t stream
) ;
#if 0
// we need this function too:
GrB_Info GB_cuda_matrix_memadvise
(
GrB_Matrix A,
what to do: advise (prefer location? access by)? prefetch? nothing?
avdice: enum (1 to 6)
int device, // GPU device or cudaCpuDeviceId
) ;
#endif
void GB_cuda_upscale_identity
(
GB_void *identity_upscaled, // output: at least sizeof (uint32_t)
GrB_Monoid monoid // input: monoid to upscale
) ;
#endif
|
