Skip to content

Commit 7d3c931

Browse files
rparolinleofang
andauthored
Update cuda_core/cuda/core/_cpp/tensor_map_cccl.h
Co-authored-by: Leo Fang <leo80042@gmail.com>
1 parent b7a956c commit 7d3c931

File tree

1 file changed

+5
-3
lines changed

1 file changed

+5
-3
lines changed

cuda_core/cuda/core/_cpp/tensor_map_cccl.h

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -5,11 +5,13 @@
55
#ifndef CUDA_CORE_TENSOR_MAP_CCCL_H_
66
#define CUDA_CORE_TENSOR_MAP_CCCL_H_
77

8-
#include <stddef.h>
9-
#include <stdint.h>
10-
118
#ifdef __cplusplus
9+
#include <cstddef>
10+
#include <cstdint>
1211
extern "C" {
12+
#else
13+
#include <stddef.h>
14+
#include <stdint.h>
1315
#endif
1416

1517
// Build a tiled CUtensorMap using CCCL's cuda::make_tma_descriptor (from <cuda/tma>).

0 commit comments

Comments
 (0)