mirror of
https://github.com/huggingface/text-generation-inference.git
synced 2025-04-21 14:52:20 +00:00
First definition of binding trtllm to rust
This commit is contained in:
parent
c6537df493
commit
4335a39f92
38
backends/trtllm/csrc/backend.cpp
Normal file
38
backends/trtllm/csrc/backend.cpp
Normal file
@ -0,0 +1,38 @@
|
|||||||
|
#include "tensorrt_llm/common/logger.h"
|
||||||
|
#include "tensorrt_llm/executor/executor.h"
|
||||||
|
#include "tensorrt_llm/plugins/api/tllmPlugin.h"
|
||||||
|
|
||||||
|
# ifdef __cplusplus
|
||||||
|
extern "C" {
|
||||||
|
#endif
|
||||||
|
|
||||||
|
#if defined __GNUC__
|
||||||
|
#define TGI_BACKEND_PUBLIC __attribute__ ((visibility("default")))
|
||||||
|
#else
|
||||||
|
#pragma message ("Compiler does not support symbol visibility.")
|
||||||
|
#define TGI_BACKEND_PUBLIC
|
||||||
|
#endif
|
||||||
|
|
||||||
|
struct tgi_trtllm_backend;
|
||||||
|
|
||||||
|
/***
|
||||||
|
* Create and allocate all the required resources to run inference over the provided engine
|
||||||
|
* @return A `tgi_trtllm_backend` handle
|
||||||
|
*/
|
||||||
|
TGI_BACKEND_PUBLIC struct tgi_trtllm_backend *tgi_trtllm_backend_create();
|
||||||
|
|
||||||
|
/***
|
||||||
|
* Free-up all the resources associated with this backend
|
||||||
|
* @param handle `tgi_trtllm_backend` handle
|
||||||
|
*/
|
||||||
|
TGI_BACKEND_PUBLIC void tgi_trtllm_backend_destroy(struct tgi_trtllm_backend *handle);
|
||||||
|
|
||||||
|
/***
|
||||||
|
*
|
||||||
|
* @param handle `tgi_trtllm_backend` handle
|
||||||
|
*/
|
||||||
|
TGI_BACKEND_PUBLIC void tgi_trtllm_backend_submit(struct tgi_trtllm_backend *handle);
|
||||||
|
|
||||||
|
#ifdef __cplusplus
|
||||||
|
};
|
||||||
|
#endif
|
Loading…
Reference in New Issue
Block a user