From 2228055bb88548837ee92a7d506dde68099d0da2 Mon Sep 17 00:00:00 2001 From: "The gemma.cpp Authors" Date: Fri, 14 Jun 2024 06:53:06 -0700 Subject: [PATCH] Internal change. PiperOrigin-RevId: 643330703 --- gemma/common.h | 48 ++++++++++++++++++++++++------------------------ 1 file changed, 24 insertions(+), 24 deletions(-) diff --git a/gemma/common.h b/gemma/common.h index e9d86f6..ff30c68 100644 --- a/gemma/common.h +++ b/gemma/common.h @@ -104,30 +104,30 @@ decltype(auto) CallForModelAndWeight(Model model, Type weight, // Used by GEMMA_EXPORT_AND_DISPATCH. For a given TWEIGHT (e.g. float), // calls FUNC> where ConfigT is chosen via MODEL enum. -#define GEMMA_DISPATCH_MODEL(MODEL, TWEIGHT, FUNC, ARGS) \ - switch (MODEL) { \ - case Model::GEMMA_TINY: { \ - HWY_EXPORT_AND_DYNAMIC_DISPATCH_T(FUNC>) \ - ARGS; \ - break; \ - } \ - case Model::GEMMA_2B: { \ - HWY_EXPORT_AND_DYNAMIC_DISPATCH_T(FUNC>) \ - ARGS; \ - break; \ - } \ - case Model::GEMMA_7B: { \ - HWY_EXPORT_AND_DYNAMIC_DISPATCH_T(FUNC>) \ - ARGS; \ - break; \ - } \ - case Model::GRIFFIN_2B: { \ - HWY_EXPORT_AND_DYNAMIC_DISPATCH_T(FUNC>) \ - ARGS; \ - break; \ - } \ - default: \ - HWY_ABORT("Model type %d unknown.", static_cast(MODEL)); \ +#define GEMMA_DISPATCH_MODEL(MODEL, TWEIGHT, FUNC, ARGS) \ + switch (MODEL) { \ + case Model::GEMMA_TINY: { \ + HWY_EXPORT_AND_DYNAMIC_DISPATCH_T(FUNC>) \ + ARGS; \ + break; \ + } \ + case Model::GEMMA_2B: { \ + HWY_EXPORT_AND_DYNAMIC_DISPATCH_T(FUNC>) \ + ARGS; \ + break; \ + } \ + case Model::GEMMA_7B: { \ + HWY_EXPORT_AND_DYNAMIC_DISPATCH_T(FUNC>) \ + ARGS; \ + break; \ + } \ + case Model::GRIFFIN_2B: { \ + HWY_EXPORT_AND_DYNAMIC_DISPATCH_T(FUNC>) \ + ARGS; \ + break; \ + } \ + default: \ + HWY_ABORT("Model type %d unknown.", static_cast(MODEL)); \ } // Like CallForModelAndWeight, but for SIMD function templates. This is a macro