intial changes

Author: keptsecret

include/nbl/builtin/hlsl/subgroup/arithmetic_portability.hlsl

@@ -9,6 +9,7 @@
 
 #include "nbl/builtin/hlsl/subgroup/basic.hlsl"
 #include "nbl/builtin/hlsl/subgroup/arithmetic_portability_impl.hlsl"
+#include "nbl/builtin/hlsl/concepts.hlsl"
 
 
 namespace nbl
@@ -25,6 +26,29 @@ struct inclusive_scan : impl::inclusive_scan<Binop,device_capabilities_traits<de
 template<class Binop, class device_capabilities=void>
 struct exclusive_scan : impl::exclusive_scan<Binop,device_capabilities_traits<device_capabilities>::shaderSubgroupArithmetic> {};
 
+}
+
+namespace subgroup2
+{
+
+template<typename Config, class BinOp, int32_t ItemsPerInvocation=1, class device_capabilities=void NBL_PRIMARY_REQUIRES(is_configuration_v<Config>)
+struct ArithmeticParams
+{
+    using config_t = Config;
+    using binop_t = BinOp;
+    using type_t = typename BinOp::type_t;
+
+    NBL_CONSTEXPR_STATIC_INLINE int32_t itemsPerInvocation = ItemsPerInvocation;
+    NBL_CONSTEXPR_STATIC_INLINE bool UseNativeIntrinsics = device_capabilities_traits<device_capabilities>::shaderSubgroupArithmetic /*&& /*some heuristic for when its faster*/;
+};
+
+template<typename Params>
+struct reduction : impl::reduction<typename Params::binop_t,Params::UseNativeIntrinsics> {};
+template<typename Params>
+struct inclusive_scan : impl::inclusive_scan<typename Params::binop_t,Params::UseNativeIntrinsics> {};
+template<typename Params>
+struct exclusive_scan : impl::exclusive_scan<typename Params::binop_t,Params::UseNativeIntrinsics> {};
+
 }
 }
 }

include/nbl/builtin/hlsl/subgroup/arithmetic_portability_impl.hlsl

@@ -140,6 +140,31 @@ struct reduction<Binop, false>
 }
 
 }
+
+namespace subgroup2
+{
+
+namespace impl
+{
+
+template<class Binop, int32_t ItemsPerInvocation, bool native>
+struct reduction
+{
+    using type_t = typename Binop::type_t;
+    using par_type_t = conditional_t<ItemsPerInvocation < 2, type_t, vector<type_t, ItemsPerInvocation> >;
+    using op_t = subgroup::impl::reduction<Binop, native>;
+
+    type_t operator()(NBL_CONST_REF_ARG(type_t) value)
+    {
+        // take the last subgroup invocation's value for the reduction
+        return BroadcastLast<type_t>(inclusive_scan<Binop,false>::__call(value));
+    }
+};
+
+}
+
+}
+
 }
 }
 

include/nbl/builtin/hlsl/subgroup/ballot.hlsl

@@ -37,6 +37,23 @@ uint32_t ElectedSubgroupInvocationID() {
     return glsl::subgroupBroadcastFirst<uint32_t>(glsl::gl_SubgroupInvocationID());
 }
 
+template<uint32_t SubgroupSizeLog2>
+struct Configuration
+{
+    using mask_t = conditional_t<SubgroupSizeLog2 < 7, conditional_t<SubgroupSizeLog2 < 6, uint32_t1, uint32_t2>, uint32_t4>;
+
+    NBL_CONSTEXPR_STATIC_INLINE uint16_t Size = 0x1u << SubgroupSizeLog2;
+};
+
+template<class T>
+struct is_configuration : bool_constant<false> {};
+
+template<uint32_t N>
+struct is_configuration<Configuration<N> > : bool_constant<true> {};
+
+template<typename T>
+NBL_CONSTEXPR bool is_configuration_v = is_configuration<T>::value;
+
 }
 }
 }