nvtx.hpp

Go to the documentation of this file.

/*
 * SPDX-FileCopyrightText: Copyright (c) 2019-2022, NVIDIA CORPORATION.
 * SPDX-License-Identifier: Apache-2.0
 */
 
#pragma once
 
#include <raft/core/nvtx.hpp>
 
namespace ML {
 
[[deprecated("Use new raft::common::nvtx::push_range from <raft/core/nvtx.hpp>")]] inline void
PUSH_RANGE(const char* name, cudaStream_t stream)
{
  raft::common::nvtx::push_range(name);
}
 
[[deprecated("Use new raft::common::nvtx::pop_range from <raft/core/nvtx.hpp>")]] inline void
POP_RANGE(cudaStream_t stream)
{
  raft::common::nvtx::pop_range();
}
 
[[deprecated("Use new raft::common::nvtx::push_range from <raft/core/nvtx.hpp>")]] inline void
PUSH_RANGE(const char* name)
{
  raft::common::nvtx::push_range(name);
}
 
[[deprecated("Use new raft::common::nvtx::pop_range from <raft/core/nvtx.hpp>")]] inline void
POP_RANGE()
{
  raft::common::nvtx::pop_range();
}
 
}  // end namespace ML