Merge pull request #287 from chillenzer/interface-takes-tag-over-acc

Allocator takes Acc by tag
alpaka-group · Feb 10, 2025 · ba2f417 · ba2f417
2 parents bde4b6f + b88d2b0
commit ba2f417
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 3 deletions.
diff --git a/examples/getAvailableSlots/source/main.cpp b/examples/getAvailableSlots/source/main.cpp
@@ -106,7 +106,7 @@ template<
 auto example03() -> int
 {
     using Allocator = mallocMC::Allocator<
-        Acc,
+        alpaka::AccToTag<Acc>,
         T_CreationPolicy,
         mallocMC::DistributionPolicies::Noop,
         mallocMC::OOMPolicies::ReturnNull,

diff --git a/examples/vectorAdd/source/main.cpp b/examples/vectorAdd/source/main.cpp
@@ -84,7 +84,7 @@ template<
 auto example01() -> int
 {
     using Allocator = mallocMC::Allocator<
-        Acc,
+        alpaka::AccToTag<Acc>,
         T_CreationPolicy,
         mallocMC::DistributionPolicies::Noop,
         mallocMC::OOMPolicies::ReturnNull,

diff --git a/include/mallocMC/allocator.hpp b/include/mallocMC/allocator.hpp
@@ -96,7 +96,7 @@ namespace mallocMC
      * @tparam T_AlignmentPolicy The desired type of a AlignmentPolicy
      */
     template<
-        typename AlpakaAcc,
+        typename T_AccTag,
         typename T_CreationPolicy,
         typename T_DistributionPolicy,
         typename T_OOMPolicy,
@@ -113,6 +113,9 @@ namespace mallocMC
         using uint32 = std::uint32_t;
 
     public:
+        using Dim = alpaka::DimInt<1>;
+        using Idx = std::uint32_t;
+        using AlpakaAcc = alpaka::TagToAcc<T_AccTag, Dim, Idx>;
         using DistributionPolicy = T_DistributionPolicy;
         using OOMPolicy = T_OOMPolicy;
         using ReservePoolPolicy = T_ReservePoolPolicy;