nvtx.hpp
Go to the documentation of this file.
1 /*
2  * Copyright (c) 2019-2022, NVIDIA CORPORATION.
3  *
4  * Licensed under the Apache License, Version 2.0 (the "License");
5  * you may not use this file except in compliance with the License.
6  * You may obtain a copy of the License at
7  *
8  * http://www.apache.org/licenses/LICENSE-2.0
9  *
10  * Unless required by applicable law or agreed to in writing, software
11  * distributed under the License is distributed on an "AS IS" BASIS,
12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13  * See the License for the specific language governing permissions and
14  * limitations under the License.
15  */
16 
17 #pragma once
18 
19 #include <raft/core/nvtx.hpp>
20 
21 namespace ML {
22 
28 [[deprecated("Use new raft::common::nvtx::push_range from <raft/core/nvtx.hpp>")]] inline void
29 PUSH_RANGE(const char* name, cudaStream_t stream)
30 {
31  raft::common::nvtx::push_range(name);
32 }
33 
38 [[deprecated("Use new raft::common::nvtx::pop_range from <raft/core/nvtx.hpp>")]] inline void
39 POP_RANGE(cudaStream_t stream)
40 {
41  raft::common::nvtx::pop_range();
42 }
43 
48 [[deprecated("Use new raft::common::nvtx::push_range from <raft/core/nvtx.hpp>")]] inline void
49 PUSH_RANGE(const char* name)
50 {
51  raft::common::nvtx::push_range(name);
52 }
53 
55 [[deprecated("Use new raft::common::nvtx::pop_range from <raft/core/nvtx.hpp>")]] inline void
57 {
58  raft::common::nvtx::pop_range();
59 }
60 
61 } // end namespace ML
Definition: dbscan.hpp:30
void POP_RANGE(cudaStream_t stream)
Synchronize CUDA stream and pop the latest nvtx range.
Definition: nvtx.hpp:39
void PUSH_RANGE(const char *name, cudaStream_t stream)
Synchronize CUDA stream and push a named nvtx range.
Definition: nvtx.hpp:29