1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
|
/*
* Copyright (c) Meta Platforms, Inc. and affiliates.
*
* This source code is licensed under the MIT license found in the
* LICENSE file in the root directory of this source tree.
*/
// -*- c++ -*-
#ifndef GPU_MACROS_IMPL_H
#define GPU_MACROS_IMPL_H
#include "../macros_impl.h"
#undef DEFINE_GETTER
#define DEFINE_GETTER(clazz, ty, name) \
ty faiss_##clazz##_##name(const Faiss##clazz* obj) { \
return static_cast<ty>( \
reinterpret_cast<const faiss::gpu::clazz*>(obj)->name); \
}
#undef DEFINE_SETTER
#define DEFINE_SETTER(clazz, ty, name) \
void faiss_##clazz##_set_##name(Faiss##clazz* obj, ty val) { \
reinterpret_cast<faiss::gpu::clazz*>(obj)->name = val; \
}
#undef DEFINE_SETTER_STATIC
#define DEFINE_SETTER_STATIC(clazz, ty_to, ty_from, name) \
void faiss_##clazz##_set_##name(Faiss##clazz* obj, ty_from val) { \
reinterpret_cast<faiss::gpu::clazz*>(obj)->name = \
static_cast<ty_to>(val); \
}
#undef DEFINE_DESTRUCTOR
#define DEFINE_DESTRUCTOR(clazz) \
void faiss_##clazz##_free(Faiss##clazz* obj) { \
delete reinterpret_cast<faiss::gpu::clazz*>(obj); \
}
#endif
|
