|
| 1 | +// Copyright 2025 Google LLC |
| 2 | +// |
| 3 | +// This source code is licensed under the BSD-style license found in the |
| 4 | +// LICENSE file in the root directory of this source tree. |
| 5 | + |
| 6 | +#include "src/operators/fingerprint_cache.h" |
| 7 | + |
| 8 | +#include <assert.h> |
| 9 | +#include <stddef.h> |
| 10 | +#include <stdint.h> |
| 11 | +#include <stdlib.h> |
| 12 | + |
| 13 | +#include "include/experimental.h" |
| 14 | +#include "include/xnnpack.h" |
| 15 | +#include "src/operators/fingerprint_id.h" |
| 16 | +#include "src/xnnpack/cache.h" |
| 17 | +#include "src/xnnpack/init-once.h" |
| 18 | +#include "src/xnnpack/mutex.h" |
| 19 | +#include "src/xnnpack/operator-type.h" |
| 20 | + |
| 21 | +#define XNN_FINGERPRINT_MAX_COUNT 256 |
| 22 | +static struct xnn_fingerprint fingerprint_vector[XNN_FINGERPRINT_MAX_COUNT]; |
| 23 | +static uint32_t fingerprint_vector_size = 0; |
| 24 | + |
| 25 | +static struct xnn_mutex mutex; |
| 26 | +XNN_INIT_ONCE_GUARD(mutex); |
| 27 | + |
| 28 | +static void init_mutex_config() { |
| 29 | + xnn_mutex_init(&mutex); |
| 30 | +} |
| 31 | + |
| 32 | +const struct xnn_fingerprint* xnn_get_fingerprint( |
| 33 | + const enum xnn_fingerprint_id id) { |
| 34 | + XNN_INIT_ONCE(mutex); |
| 35 | + uint32_t i = 0; |
| 36 | + xnn_mutex_lock(&mutex); |
| 37 | + for (; i < fingerprint_vector_size; ++i) { |
| 38 | + if (fingerprint_vector[i].id == id) { |
| 39 | + break; |
| 40 | + } |
| 41 | + } |
| 42 | + xnn_mutex_unlock(&mutex); |
| 43 | + return i < fingerprint_vector_size ? fingerprint_vector + i : NULL; |
| 44 | +} |
| 45 | + |
| 46 | +void xnn_set_fingerprint(const struct xnn_fingerprint fingerprint) { |
| 47 | + XNN_INIT_ONCE(mutex); |
| 48 | + uint32_t i = 0; |
| 49 | + xnn_mutex_lock(&mutex); |
| 50 | + for (; i < fingerprint_vector_size; ++i) { |
| 51 | + if (fingerprint_vector[i].id == fingerprint.id) { |
| 52 | + fingerprint_vector[i] = fingerprint; |
| 53 | + return; |
| 54 | + } |
| 55 | + } |
| 56 | + xnn_mutex_unlock(&mutex); |
| 57 | + if (i < fingerprint_vector_size) { |
| 58 | + return; |
| 59 | + } |
| 60 | + assert(fingerprint_vector_size < XNN_FINGERPRINT_MAX_COUNT); |
| 61 | + fingerprint_vector[fingerprint_vector_size++] = fingerprint; |
| 62 | +} |
| 63 | + |
| 64 | +void xnn_clear_fingerprints() { |
| 65 | + XNN_INIT_ONCE(mutex); |
| 66 | + xnn_mutex_lock(&mutex); |
| 67 | + fingerprint_vector_size = 0; |
| 68 | + xnn_mutex_unlock(&mutex); |
| 69 | +} |
| 70 | + |
| 71 | +#define XNNPACK_OP_TYPE_TO_FINGERPRINT(name) \ |
| 72 | + case xnn_operator_type_##name: \ |
| 73 | + return xnn_fingerprint_id_##name; |
| 74 | + |
| 75 | +enum xnn_fingerprint_id fingerprint_id_from_operator_type( |
| 76 | + const enum xnn_operator_type operator_type) { |
| 77 | + // LINT.IfChange(operator_to_fingerprint_id) |
| 78 | + switch (operator_type) { |
| 79 | + default: |
| 80 | + return xnn_fingerprint_id_unknown; |
| 81 | + } |
| 82 | + // LINT.ThenChange(fingerprint_id.h:fingerprint_id) |
| 83 | +} |
| 84 | + |
| 85 | +#undef XNNPACK_OP_TYPE_TO_FINGERPRINT |
| 86 | + |
| 87 | +// The context for an XNNPack weight cache provider that we pass to operator |
| 88 | +// `create` functions when we want to fingerprint them. |
| 89 | +struct fingerprint_cache_context { |
| 90 | + void* buffer; |
| 91 | + size_t bytes; |
| 92 | + uint32_t hash; |
| 93 | +}; |
| 94 | + |
| 95 | +static size_t fingerprint_cache_look_up( |
| 96 | + void* context, const struct xnn_weights_cache_look_up_key* cache_key) { |
| 97 | + return XNN_CACHE_NOT_FOUND; |
| 98 | +} |
| 99 | + |
| 100 | +static void* fingerprint_cache_reserve_space(void* const context, size_t n) { |
| 101 | + struct fingerprint_cache_context* const ctx = context; |
| 102 | + void* buffer = realloc(ctx->buffer, n); |
| 103 | + if (!buffer) { |
| 104 | + return NULL; |
| 105 | + } |
| 106 | + ctx->buffer = buffer; |
| 107 | + ctx->bytes = n; |
| 108 | + return ctx->buffer; |
| 109 | +} |
| 110 | + |
| 111 | +static size_t fingerprint_cache_look_up_or_insert( |
| 112 | + void* context, const struct xnn_weights_cache_look_up_key* cache_key, |
| 113 | + void* ptr, size_t size) { |
| 114 | + assert(context); |
| 115 | + struct fingerprint_cache_context* const ctx = context; |
| 116 | + ctx->hash = murmur_hash3(ptr, size, /*seed=*/ctx->hash); |
| 117 | + return 0; |
| 118 | +} |
| 119 | + |
| 120 | +static bool fingerprint_cache_is_finalized(void* context) { return false; } |
| 121 | + |
| 122 | +static void* fingerprint_cache_offset_to_addr(void* context, size_t offset) { |
| 123 | + assert(context); |
| 124 | + struct fingerprint_cache_context* const ctx = context; |
| 125 | + return ctx->buffer; |
| 126 | +} |
| 127 | + |
| 128 | +static enum xnn_status fingerprint_cache_delete_cache(void* context) { |
| 129 | + struct fingerprint_cache_context* const ctx = context; |
| 130 | + if (ctx) { |
| 131 | + if (ctx->buffer) { |
| 132 | + free(ctx->buffer); |
| 133 | + } |
| 134 | + *ctx = (struct fingerprint_cache_context){0}; |
| 135 | + } |
| 136 | + return xnn_status_success; |
| 137 | +} |
| 138 | + |
| 139 | +struct fingerprint_context create_fingerprint_context_for_op( |
| 140 | + const enum xnn_operator_type operator_type) { |
| 141 | + return create_fingerprint_context( |
| 142 | + fingerprint_id_from_operator_type(operator_type)); |
| 143 | +} |
| 144 | + |
| 145 | +struct fingerprint_context create_fingerprint_context( |
| 146 | + const enum xnn_fingerprint_id fingerprint_id) { |
| 147 | + struct fingerprint_context context = { |
| 148 | + .status = xnn_status_uninitialized, |
| 149 | + .fingerprint_id = fingerprint_id, |
| 150 | + .cache = |
| 151 | + (struct xnn_weights_cache_provider){ |
| 152 | + .context = NULL, |
| 153 | + .look_up = fingerprint_cache_look_up, |
| 154 | + .reserve_space = fingerprint_cache_reserve_space, |
| 155 | + .look_up_or_insert = fingerprint_cache_look_up_or_insert, |
| 156 | + .is_finalized = fingerprint_cache_is_finalized, |
| 157 | + .offset_to_addr = fingerprint_cache_offset_to_addr, |
| 158 | + .delete_cache = fingerprint_cache_delete_cache}, |
| 159 | + }; |
| 160 | + if (context.fingerprint_id == xnn_fingerprint_id_unknown) { |
| 161 | + context.status = xnn_status_unsupported_parameter; |
| 162 | + } else if (xnn_get_fingerprint(context.fingerprint_id)) { |
| 163 | + context.status = xnn_status_success; |
| 164 | + } else { |
| 165 | + // Do this after the checks to avoid a memory allocation when unnecessary. |
| 166 | + context.cache.context = calloc(1, sizeof(struct fingerprint_cache_context)); |
| 167 | + } |
| 168 | + return context; |
| 169 | +} |
| 170 | + |
| 171 | +static void free_fingerprint_cache_provider( |
| 172 | + struct xnn_weights_cache_provider* const provider) { |
| 173 | + if (provider) { |
| 174 | + provider->delete_cache(provider->context); |
| 175 | + if (provider->context) { |
| 176 | + free(provider->context); |
| 177 | + } |
| 178 | + } |
| 179 | +} |
| 180 | + |
| 181 | +void finalize_fingerprint_context(struct fingerprint_context* const context) { |
| 182 | + assert(context); |
| 183 | + if (context->status == xnn_status_uninitialized) { |
| 184 | + xnn_set_fingerprint((struct xnn_fingerprint){ |
| 185 | + .id = context->fingerprint_id, |
| 186 | + .value = fingerprint_cache_get_fingerprint(&context->cache)}); |
| 187 | + } |
| 188 | + free_fingerprint_cache_provider(&context->cache); |
| 189 | +} |
| 190 | + |
| 191 | +uint32_t fingerprint_cache_get_fingerprint( |
| 192 | + const struct xnn_weights_cache_provider* const provider) { |
| 193 | + assert(provider); |
| 194 | + assert(provider->context); |
| 195 | + return ((struct fingerprint_cache_context*)provider->context)->hash; |
| 196 | +} |
0 commit comments