Fixed base operator types

NVIDIA · Aug 8, 2024 · 3b57bf2 · 3b57bf2
1 parent 39b4372
commit 3b57bf2
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 4 deletions.
diff --git a/include/matx/operators/at.h b/include/matx/operators/at.h
@@ -45,7 +45,7 @@ namespace matx
     class AtOp : public BaseOp<AtOp<Op, Is...>>
     {
       private:
-        Op op_;
+        typename base_type<Op>::type op_;
         cuda::std::array<index_t, sizeof...(Is)> idx_;
 
       public:

diff --git a/include/matx/operators/concat.h b/include/matx/operators/concat.h
@@ -227,7 +227,7 @@ namespace matx
       }
 
       private:
-      cuda::std::tuple<Ts...> ops_;
+      cuda::std::tuple<typename base_type<Ts>::type ...> ops_;
       index_t size_;    
       int axis_;
     }; // end class ConcatOp

diff --git a/include/matx/operators/frexp.h b/include/matx/operators/frexp.h
@@ -44,7 +44,7 @@ namespace detail {
   class FrexpOp : public BaseOp<FrexpOp<OpA, WHICH>>
   {
     private:
-      OpA a_;
+      typename base_type<OpA>::type a_;
 
     public:
       using matxop = bool;

diff --git a/include/matx/operators/stack.h b/include/matx/operators/stack.h
@@ -199,7 +199,7 @@ namespace matx
       }
 
       private:
-      cuda::std::tuple<Ts...> ops_;
+      cuda::std::tuple<typename base_type<Ts>::type ...> ops_;
       index_t size_;    
       int axis_;
     }; // end class StackOp