mirror of
https://github.com/microsoft/mscclpp.git
synced 2026-05-12 09:17:06 +00:00
a707273701
Reorganize current native algorithm implementation and DSL algorithm implementation. Provide unified API for DSL algo and native algo and provide interface to tune the algo Provide interface for pytorch integration with native API and DSL --------- Co-authored-by: Copilot <175728472+Copilot@users.noreply.github.com> Co-authored-by: Copilot <198982749+Copilot@users.noreply.github.com> Co-authored-by: chhwang <8018170+chhwang@users.noreply.github.com>
21 lines
588 B
C++
21 lines
588 B
C++
// Copyright (c) Microsoft Corporation.
|
|
// Licensed under the MIT license.
|
|
|
|
#ifndef MSCCLPP_ATOMIC_HPP_
|
|
#define MSCCLPP_ATOMIC_HPP_
|
|
|
|
#if defined(MSCCLPP_USE_CUDA)
|
|
#ifndef MSCCLPP_DEVICE_CUDA
|
|
#define MSCCLPP_DEVICE_CUDA
|
|
#include <mscclpp/atomic_device.hpp>
|
|
#undef MSCCLPP_DEVICE_CUDA
|
|
#endif // !defined(MSCCLPP_DEVICE_CUDA)
|
|
#else // !defined(MSCCLPP_USE_CUDA)
|
|
#ifndef MSCCLPP_DEVICE_HIP
|
|
#define MSCCLPP_DEVICE_HIP
|
|
#include <mscclpp/atomic_device.hpp>
|
|
#undef MSCCLPP_DEVICE_HIP
|
|
#endif // !defined(MSCCLPP_DEVICE_HIP)
|
|
#endif // !defined(MSCCLPP_USE_CUDA)
|
|
|
|
#endif // MSCCLPP_ATOMIC_HPP_
|