Welcome to mirror list, hosted at ThFree Co, Russian Federation.

github.com/marian-nmt/intgemm/intgemm.git - Unnamed repository; edit this file 'description' to name the repository.
summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
Diffstat (limited to 'callbacks/configs.h')
-rw-r--r--callbacks/configs.h7
1 files changed, 7 insertions, 0 deletions
diff --git a/callbacks/configs.h b/callbacks/configs.h
index 4a01e2e..8e2eacc 100644
--- a/callbacks/configs.h
+++ b/callbacks/configs.h
@@ -20,6 +20,13 @@ struct UnquantizeAndWrite {
UnquantizeAndWrite(float unquant_mult, float* addr) : unquant_mult(unquant_mult), addr(addr) {}
};
+struct AddBiasAndWrite {
+ const int* bias_addr;
+ int* output_addr;
+
+ AddBiasAndWrite(const int* bias_addr, int* output_addr) : bias_addr(bias_addr), output_addr(output_addr) {}
+};
+
struct UnquantizeAndAddBiasAndWrite {
float unquant_mult;
const float* bias_addr;