Skip to content

Commit 0a29aea

Browse files
committed
refactor: Remove Buffers class and associated bindings from pybind and nanobind
- Deleted Buffers class and its bindings from both pybind and nanobind implementations. - Cleaned up include statements in related files to streamline the codebase. Signed-off-by: Robin Kobus <[email protected]>
1 parent 7f56208 commit 0a29aea

File tree

6 files changed

+0
-140
lines changed

6 files changed

+0
-140
lines changed

cpp/tensorrt_llm/nanobind/batch_manager/bindings.cpp

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -23,7 +23,6 @@
2323
#include "tensorrt_llm/batch_manager/microBatchScheduler.h"
2424
#include "tensorrt_llm/batch_manager/peftCacheManager.h"
2525
#include "tensorrt_llm/batch_manager/rnnStateManager.h"
26-
#include "tensorrt_llm/batch_manager/runtimeBuffers.h"
2726
#include "tensorrt_llm/batch_manager/sequenceSlotManager.h"
2827
#include "tensorrt_llm/nanobind/common/bindTypes.h"
2928
#include "tensorrt_llm/runtime/gptDecoderBatched.h"

cpp/tensorrt_llm/pybind/CMakeLists.txt

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,6 @@ set(TRTLLM_PYBIND_MODULE
66
set(SRCS
77
batch_manager/algorithms.cpp
88
batch_manager/bindings.cpp
9-
batch_manager/buffers.cpp
109
batch_manager/cacheTransceiver.cpp
1110
batch_manager/kvCacheManager.cpp
1211
batch_manager/llmRequest.cpp

cpp/tensorrt_llm/pybind/batch_manager/bindings.cpp

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -22,7 +22,6 @@
2222
#include "tensorrt_llm/batch_manager/microBatchScheduler.h"
2323
#include "tensorrt_llm/batch_manager/peftCacheManager.h"
2424
#include "tensorrt_llm/batch_manager/rnnStateManager.h"
25-
#include "tensorrt_llm/batch_manager/runtimeBuffers.h"
2625
#include "tensorrt_llm/batch_manager/sequenceSlotManager.h"
2726
#include "tensorrt_llm/pybind/common/bindTypes.h"
2827
#include "tensorrt_llm/runtime/gptDecoderBatched.h"

cpp/tensorrt_llm/pybind/batch_manager/buffers.cpp

Lines changed: 0 additions & 105 deletions
This file was deleted.

cpp/tensorrt_llm/pybind/batch_manager/buffers.h

Lines changed: 0 additions & 30 deletions
This file was deleted.

cpp/tensorrt_llm/pybind/bindings.cpp

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -27,7 +27,6 @@
2727
#include "tensorrt_llm/common/quantization.h"
2828
#include "tensorrt_llm/pybind/batch_manager/algorithms.h"
2929
#include "tensorrt_llm/pybind/batch_manager/bindings.h"
30-
#include "tensorrt_llm/pybind/batch_manager/buffers.h"
3130
#include "tensorrt_llm/pybind/batch_manager/cacheTransceiver.h"
3231
#include "tensorrt_llm/pybind/batch_manager/kvCacheManager.h"
3332
#include "tensorrt_llm/pybind/batch_manager/llmRequest.h"
@@ -467,7 +466,6 @@ PYBIND11_MODULE(TRTLLM_PYBIND_MODULE, m)
467466
tb::kv_cache_manager::KVCacheManagerBindings::initBindings(mInternalBatchManager);
468467
tb::BasePeftCacheManagerBindings::initBindings(mInternalBatchManager);
469468
tb::CacheTransceiverBindings::initBindings(mInternalBatchManager);
470-
tpb::Buffers::initBindings(mInternalBatchManager);
471469

472470
auto mInternalAlgorithms = mInternal.def_submodule("algorithms", "Algorithms internal bindings");
473471
tpb::algorithms::initBindings(mInternalAlgorithms);

0 commit comments

Comments
 (0)