Remove thneed logic from snpemodel

old-commit-hash: ef8e0fef55
beeps
mitchellgoffpc 2 years ago
parent a3fbbb26ac
commit 68e2c74cd3
  1. 6
      selfdrive/modeld/runners/snpemodel.cc
  2. 9
      selfdrive/modeld/runners/snpemodel.h

@ -75,12 +75,6 @@ SNPEModel::SNPEModel(const std::string path, float *_output, size_t _output_size
std::vector<size_t> output_strides = {output_size * sizeof(float), sizeof(float)};
output_buffer = ub_factory.createUserBuffer(output, output_size * sizeof(float), output_strides, &ub_encoding_float);
output_map.add(output_tensor_name, output_buffer.get());
#ifdef USE_THNEED
if (snpe_runtime == zdl::DlSystem::Runtime_t::GPU) {
thneed.reset(new Thneed());
}
#endif
}
void SNPEModel::addInput(const std::string name, float *buffer, int size) {

@ -17,10 +17,6 @@
#include "selfdrive/modeld/runners/runmodel.h"
#ifdef USE_THNEED
#include "selfdrive/modeld/thneed/thneed.h"
#endif
struct SNPEModelInput : public ModelInput {
std::unique_ptr<zdl::DlSystem::IUserBuffer> snpe_buffer;
@ -37,11 +33,6 @@ public:
void addInput(const std::string name, float *buffer, int size);
void execute();
#ifdef USE_THNEED
std::unique_ptr<Thneed> thneed;
bool thneed_recorded = false;
#endif
private:
std::string model_data;

Loading…
Cancel
Save