cuda-cccl 0.3.3__cp313-cp313-win_amd64.whl

cuda/cccl/headers/include/thrust/detail/gather.inl

@@ -0,0 +1,173 @@
+/*
+ *  Copyright 2008-2013 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 <thrust/detail/config.h>
+
+#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 <thrust/gather.h>
+#include <thrust/iterator/iterator_traits.h>
+#include <thrust/system/detail/adl/gather.h>
+#include <thrust/system/detail/generic/gather.h>
+#include <thrust/system/detail/generic/select_system.h>
+
+THRUST_NAMESPACE_BEGIN
+
+_CCCL_EXEC_CHECK_DISABLE
+template <typename DerivedPolicy, typename InputIterator, typename RandomAccessIterator, typename OutputIterator>
+_CCCL_HOST_DEVICE OutputIterator gather(
+  const thrust::detail::execution_policy_base<DerivedPolicy>& exec,
+  InputIterator map_first,
+  InputIterator map_last,
+  RandomAccessIterator input_first,
+  OutputIterator result)
+{
+  _CCCL_NVTX_RANGE_SCOPE("thrust::gather");
+  using thrust::system::detail::generic::gather;
+  return gather(
+    thrust::detail::derived_cast(thrust::detail::strip_const(exec)), map_first, map_last, input_first, result);
+} // end gather()
+
+_CCCL_EXEC_CHECK_DISABLE
+template <typename DerivedPolicy,
+          typename InputIterator1,
+          typename InputIterator2,
+          typename RandomAccessIterator,
+          typename OutputIterator>
+_CCCL_HOST_DEVICE OutputIterator gather_if(
+  const thrust::detail::execution_policy_base<DerivedPolicy>& exec,
+  InputIterator1 map_first,
+  InputIterator1 map_last,
+  InputIterator2 stencil,
+  RandomAccessIterator input_first,
+  OutputIterator result)
+{
+  _CCCL_NVTX_RANGE_SCOPE("thrust::gather_if");
+  using thrust::system::detail::generic::gather_if;
+  return gather_if(
+    thrust::detail::derived_cast(thrust::detail::strip_const(exec)), map_first, map_last, stencil, input_first, result);
+} // end gather_if()
+
+_CCCL_EXEC_CHECK_DISABLE
+template <typename DerivedPolicy,
+          typename InputIterator1,
+          typename InputIterator2,
+          typename RandomAccessIterator,
+          typename OutputIterator,
+          typename Predicate>
+_CCCL_HOST_DEVICE OutputIterator gather_if(
+  const thrust::detail::execution_policy_base<DerivedPolicy>& exec,
+  InputIterator1 map_first,
+  InputIterator1 map_last,
+  InputIterator2 stencil,
+  RandomAccessIterator input_first,
+  OutputIterator result,
+  Predicate pred)
+{
+  _CCCL_NVTX_RANGE_SCOPE("thrust::gather_if");
+  using thrust::system::detail::generic::gather_if;
+  return gather_if(
+    thrust::detail::derived_cast(thrust::detail::strip_const(exec)),
+    map_first,
+    map_last,
+    stencil,
+    input_first,
+    result,
+    pred);
+} // end gather_if()
+
+template <typename InputIterator, typename RandomAccessIterator, typename OutputIterator>
+OutputIterator
+gather(InputIterator map_first, InputIterator map_last, RandomAccessIterator input_first, OutputIterator result)
+{
+  _CCCL_NVTX_RANGE_SCOPE("thrust::gather");
+  using thrust::system::detail::generic::select_system;
+
+  using System1 = typename thrust::iterator_system<InputIterator>::type;
+  using System2 = typename thrust::iterator_system<RandomAccessIterator>::type;
+  using System3 = typename thrust::iterator_system<OutputIterator>::type;
+
+  System1 system1;
+  System2 system2;
+  System3 system3;
+
+  return thrust::gather(select_system(system1, system2, system3), map_first, map_last, input_first, result);
+} // end gather()
+
+template <typename InputIterator1, typename InputIterator2, typename RandomAccessIterator, typename OutputIterator>
+OutputIterator gather_if(
+  InputIterator1 map_first,
+  InputIterator1 map_last,
+  InputIterator2 stencil,
+  RandomAccessIterator input_first,
+  OutputIterator result)
+{
+  _CCCL_NVTX_RANGE_SCOPE("thrust::gather_if");
+  using thrust::system::detail::generic::select_system;
+
+  using System1 = typename thrust::iterator_system<InputIterator1>::type;
+  using System2 = typename thrust::iterator_system<InputIterator2>::type;
+  using System3 = typename thrust::iterator_system<RandomAccessIterator>::type;
+  using System4 = typename thrust::iterator_system<OutputIterator>::type;
+
+  System1 system1;
+  System2 system2;
+  System3 system3;
+  System4 system4;
+
+  return thrust::gather_if(
+    select_system(system1, system2, system3, system4), map_first, map_last, stencil, input_first, result);
+} // end gather_if()
+
+template <typename InputIterator1,
+          typename InputIterator2,
+          typename RandomAccessIterator,
+          typename OutputIterator,
+          typename Predicate>
+OutputIterator gather_if(
+  InputIterator1 map_first,
+  InputIterator1 map_last,
+  InputIterator2 stencil,
+  RandomAccessIterator input_first,
+  OutputIterator result,
+  Predicate pred)
+{
+  _CCCL_NVTX_RANGE_SCOPE("thrust::gather_if");
+  using thrust::system::detail::generic::select_system;
+
+  using System1 = typename thrust::iterator_system<InputIterator1>::type;
+  using System2 = typename thrust::iterator_system<InputIterator2>::type;
+  using System3 = typename thrust::iterator_system<RandomAccessIterator>::type;
+  using System4 = typename thrust::iterator_system<OutputIterator>::type;
+
+  System1 system1;
+  System2 system2;
+  System3 system3;
+  System4 system4;
+
+  return thrust::gather_if(
+    select_system(system1, system2, system3, system4), map_first, map_last, stencil, input_first, result, pred);
+} // end gather_if()
+
+THRUST_NAMESPACE_END

cuda/cccl/headers/include/thrust/detail/generate.inl

@@ -0,0 +1,86 @@
+/*
+ *  Copyright 2008-2013 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 <thrust/detail/config.h>
+
+#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 <thrust/generate.h>
+#include <thrust/iterator/iterator_traits.h>
+#include <thrust/system/detail/adl/generate.h>
+#include <thrust/system/detail/generic/generate.h>
+#include <thrust/system/detail/generic/select_system.h>
+
+THRUST_NAMESPACE_BEGIN
+
+_CCCL_EXEC_CHECK_DISABLE
+template <typename DerivedPolicy, typename ForwardIterator, typename Generator>
+_CCCL_HOST_DEVICE void
+generate(const thrust::detail::execution_policy_base<DerivedPolicy>& exec,
+         ForwardIterator first,
+         ForwardIterator last,
+         Generator gen)
+{
+  _CCCL_NVTX_RANGE_SCOPE("thrust::generate");
+  using thrust::system::detail::generic::generate;
+  return generate(thrust::detail::derived_cast(thrust::detail::strip_const(exec)), first, last, gen);
+} // end generate()
+
+_CCCL_EXEC_CHECK_DISABLE
+template <typename DerivedPolicy, typename OutputIterator, typename Size, typename Generator>
+_CCCL_HOST_DEVICE OutputIterator generate_n(
+  const thrust::detail::execution_policy_base<DerivedPolicy>& exec, OutputIterator first, Size n, Generator gen)
+{
+  _CCCL_NVTX_RANGE_SCOPE("thrust::generate_n");
+  using thrust::system::detail::generic::generate_n;
+  return generate_n(thrust::detail::derived_cast(thrust::detail::strip_const(exec)), first, n, gen);
+} // end generate_n()
+
+template <typename ForwardIterator, typename Generator>
+void generate(ForwardIterator first, ForwardIterator last, Generator gen)
+{
+  _CCCL_NVTX_RANGE_SCOPE("thrust::generate");
+  using thrust::system::detail::generic::select_system;
+
+  using System = typename thrust::iterator_system<ForwardIterator>::type;
+
+  System system;
+
+  return thrust::generate(select_system(system), first, last, gen);
+} // end generate()
+
+template <typename OutputIterator, typename Size, typename Generator>
+OutputIterator generate_n(OutputIterator first, Size n, Generator gen)
+{
+  _CCCL_NVTX_RANGE_SCOPE("thrust::generate_n");
+  using thrust::system::detail::generic::select_system;
+
+  using System = typename thrust::iterator_system<OutputIterator>::type;
+
+  System system;
+
+  return thrust::generate_n(select_system(system), first, n, gen);
+} // end generate_n()
+
+THRUST_NAMESPACE_END

cuda/cccl/headers/include/thrust/detail/get_iterator_value.h

@@ -0,0 +1,62 @@
+/*
+ *  Copyright 2008-2016 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 <thrust/detail/config.h>
+
+#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 <thrust/detail/type_traits/pointer_traits.h>
+#include <thrust/execution_policy.h>
+#include <thrust/iterator/iterator_traits.h>
+#include <thrust/system/detail/generic/memory.h> // for get_value()
+
+THRUST_NAMESPACE_BEGIN
+
+namespace detail
+{
+
+// get_iterator_value specialization on iterators
+// --------------------------------------------------
+// it is okay to dereference iterator in the usual way
+template <typename DerivedPolicy, typename Iterator>
+_CCCL_HOST_DEVICE it_value_t<Iterator> get_iterator_value(thrust::execution_policy<DerivedPolicy>&, Iterator it)
+{
+  return *it;
+} // get_iterator_value(exec,Iterator);
+
+// get_iterator_value specialization on pointer
+// ----------------------------------------------
+// we can't just dereference a pointer in the usual way, because
+// it may point to a location in the device memory.
+// we use get_value(exec,pointer*) function
+// to perform a dereferencing consistent with the execution policy
+template <typename DerivedPolicy, typename Pointer>
+_CCCL_HOST_DEVICE typename thrust::detail::pointer_traits<Pointer*>::element_type
+get_iterator_value(thrust::execution_policy<DerivedPolicy>& exec, Pointer* ptr)
+{
+  return get_value(derived_cast(exec), ptr);
+} // get_iterator_value(exec,Pointer*)
+
+} // namespace detail
+
+THRUST_NAMESPACE_END

cuda/cccl/headers/include/thrust/detail/inner_product.inl

@@ -0,0 +1,118 @@
+/*
+ *  Copyright 2008-2013 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 <thrust/detail/config.h>
+
+#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 <thrust/inner_product.h>
+#include <thrust/iterator/iterator_traits.h>
+#include <thrust/system/detail/adl/inner_product.h>
+#include <thrust/system/detail/generic/inner_product.h>
+#include <thrust/system/detail/generic/select_system.h>
+
+THRUST_NAMESPACE_BEGIN
+
+_CCCL_EXEC_CHECK_DISABLE
+template <typename DerivedPolicy, typename InputIterator1, typename InputIterator2, typename OutputType>
+_CCCL_HOST_DEVICE OutputType inner_product(
+  const thrust::detail::execution_policy_base<DerivedPolicy>& exec,
+  InputIterator1 first1,
+  InputIterator1 last1,
+  InputIterator2 first2,
+  OutputType init)
+{
+  _CCCL_NVTX_RANGE_SCOPE("thrust::inner_product");
+  using thrust::system::detail::generic::inner_product;
+  return inner_product(thrust::detail::derived_cast(thrust::detail::strip_const(exec)), first1, last1, first2, init);
+} // end inner_product()
+
+_CCCL_EXEC_CHECK_DISABLE
+template <typename DerivedPolicy,
+          typename InputIterator1,
+          typename InputIterator2,
+          typename OutputType,
+          typename BinaryFunction1,
+          typename BinaryFunction2>
+_CCCL_HOST_DEVICE OutputType inner_product(
+  const thrust::detail::execution_policy_base<DerivedPolicy>& exec,
+  InputIterator1 first1,
+  InputIterator1 last1,
+  InputIterator2 first2,
+  OutputType init,
+  BinaryFunction1 binary_op1,
+  BinaryFunction2 binary_op2)
+{
+  _CCCL_NVTX_RANGE_SCOPE("thrust::inner_product");
+  using thrust::system::detail::generic::inner_product;
+  return inner_product(
+    thrust::detail::derived_cast(thrust::detail::strip_const(exec)),
+    first1,
+    last1,
+    first2,
+    init,
+    binary_op1,
+    binary_op2);
+} // end inner_product()
+
+template <typename InputIterator1, typename InputIterator2, typename OutputType>
+OutputType inner_product(InputIterator1 first1, InputIterator1 last1, InputIterator2 first2, OutputType init)
+{
+  _CCCL_NVTX_RANGE_SCOPE("thrust::inner_product");
+  using thrust::system::detail::generic::select_system;
+
+  using System1 = typename thrust::iterator_system<InputIterator1>::type;
+  using System2 = typename thrust::iterator_system<InputIterator2>::type;
+
+  System1 system1;
+  System2 system2;
+
+  return thrust::inner_product(select_system(system1, system2), first1, last1, first2, init);
+} // end inner_product()
+
+template <typename InputIterator1,
+          typename InputIterator2,
+          typename OutputType,
+          typename BinaryFunction1,
+          typename BinaryFunction2>
+OutputType inner_product(
+  InputIterator1 first1,
+  InputIterator1 last1,
+  InputIterator2 first2,
+  OutputType init,
+  BinaryFunction1 binary_op1,
+  BinaryFunction2 binary_op2)
+{
+  _CCCL_NVTX_RANGE_SCOPE("thrust::inner_product");
+  using thrust::system::detail::generic::select_system;
+
+  using System1 = typename thrust::iterator_system<InputIterator1>::type;
+  using System2 = typename thrust::iterator_system<InputIterator2>::type;
+
+  System1 system1;
+  System2 system2;
+
+  return thrust::inner_product(select_system(system1, system2), first1, last1, first2, init, binary_op1, binary_op2);
+} // end inner_product()
+
+THRUST_NAMESPACE_END