Add missing parameter

d93e6486 · rocking · 2a69b39d · d93e6486
Commit d93e6486 authored Mar 20, 2023 by rocking
Show whitespace changes
Inline Side-by-side

Showing with 4 additions and 1 deletion

include/ck/tensor_operation/gpu/element/quantization_operation.hpp ...k/tensor_operation/gpu/element/quantization_operation.hpp +4 -1

No files found.
--- a/include/ck/tensor_operation/gpu/element/quantization_operation.hpp
+++ b/include/ck/tensor_operation/gpu/element/quantization_operation.hpp
@@ -246,7 +246,10 @@ struct Add_Mul_Activation_Mul_Clamp
 template <typename Activation>
 struct Add_Mul2_Activation_Mul_Clamp
 {
-    Add_Mul2_Activation_Mul_Clamp(Activation activationOp) : activationOp_(activationOp) {}
+    Add_Mul2_Activation_Mul_Clamp(float scale_z_inv, Activation activationOp)
+        : scale_z_inv_(scale_z_inv), activationOp_(activationOp)
+    {
+    }
    __host__ __device__ constexpr void
    operator()(int8_t& y, const int32_t& x, const int32_t& bias, const float& scaleAcc) const