Skip to content

Commit

Permalink
add destroy
Browse files Browse the repository at this point in the history
  • Loading branch information
tikikun authored and hiento09 committed Mar 15, 2024
1 parent 9db2a5e commit fb0ca31
Show file tree
Hide file tree
Showing 3 changed files with 31 additions and 1 deletion.
11 changes: 11 additions & 0 deletions cpp/tensorrt_llm/nitro/controllers/processManager.cc
Original file line number Diff line number Diff line change
@@ -0,0 +1,11 @@
#include "processManager.h"
#include <cstdlib>
#include <trantor/utils/Logger.h>

void processManager::destroy(
const HttpRequestPtr &req,
std::function<void(const HttpResponsePtr &)> &&callback) {
LOG_INFO << "Program is exitting, goodbye!";
exit(0);
return;
};
19 changes: 19 additions & 0 deletions cpp/tensorrt_llm/nitro/controllers/processManager.h
Original file line number Diff line number Diff line change
@@ -0,0 +1,19 @@
#pragma once

#include <drogon/HttpController.h>
#include <drogon/HttpTypes.h>

using namespace drogon;

class processManager : public drogon::HttpController<processManager> {
public:
METHOD_LIST_BEGIN

METHOD_ADD(processManager::destroy, "/destroy",
Delete); // path is /processManager/{arg1}/{arg2}/list

METHOD_LIST_END

void destroy(const HttpRequestPtr &req,
std::function<void(const HttpResponsePtr &)> &&callback);
};
2 changes: 1 addition & 1 deletion cpp/tensorrt_llm/nitro/main.cc
Original file line number Diff line number Diff line change
Expand Up @@ -49,7 +49,7 @@ int main(int argc, char* argv[])
}

int logical_cores = std::thread::hardware_concurrency();
int drogon_thread_num = 1; // temporarily set thread num to 1
int drogon_thread_num = std::thread::hardware_concurrency(); // temporarily set thread num to 1
nitro_utils::nitro_logo();
#ifdef NITRO_VERSION
LOG_INFO << "Nitro version: " << NITRO_VERSION;
Expand Down

0 comments on commit fb0ca31

Please sign in to comment.