/* * Copyright 2008-2018 NVIDIA Corporation * * Licensed under the Apache License, Version 2.0 (the "License"); * you may not use this file except in compliance with the License. * You may obtain a copy of the License at * * http://www.apache.org/licenses/LICENSE-2.0 * * Unless required by applicable law or agreed to in writing, software * distributed under the License is distributed on an "AS IS" BASIS, * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. * See the License for the specific language governing permissions and * limitations under the License. */ #pragma once #include #if defined(_CCCL_IMPLICIT_SYSTEM_HEADER_GCC) # pragma GCC system_header #elif defined(_CCCL_IMPLICIT_SYSTEM_HEADER_CLANG) # pragma clang system_header #elif defined(_CCCL_IMPLICIT_SYSTEM_HEADER_MSVC) # pragma system_header #endif // no system header #include #include #include THRUST_NAMESPACE_BEGIN namespace system { namespace cpp { pointer malloc(std::size_t n) { tag t; return pointer(thrust::system::detail::sequential::malloc(t, n)); } // end malloc() template pointer malloc(std::size_t n) { pointer raw_ptr = thrust::system::cpp::malloc(sizeof(T) * n); return pointer(reinterpret_cast(raw_ptr.get())); } // end malloc() void free(pointer ptr) { tag t; return thrust::system::detail::sequential::free(t, ptr); } // end free() } // namespace cpp } // namespace system THRUST_NAMESPACE_END