Skip to content

Commit 5db928d

Browse files
committed
Fix minor things
Signed-off-by: Dheeraj Peri <[email protected]>
1 parent 18f867d commit 5db928d

File tree

2 files changed

+1
-24
lines changed

2 files changed

+1
-24
lines changed

examples/sample_rt_app/README.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -32,5 +32,5 @@ To build and run the app
3232

3333
```
3434
cd TRTorch
35-
bs run //examples/sample_rt_app:samplertapp $PWD/examples/sample_rt_app/norm.jit
35+
bazel run //examples/sample_rt_app:samplertapp $PWD/examples/sample_rt_app/norm.jit
3636
```

examples/sample_rt_app/main.cpp

Lines changed: 0 additions & 23 deletions
Original file line numberDiff line numberDiff line change
@@ -5,28 +5,6 @@
55
#include <vector>
66
#include "torch/script.h"
77

8-
// Load the TRT engine from engine_path
9-
std::vector<char> loadEngine(std::string engine_path){
10-
std::ifstream engineFile(engine_path, std::ios::binary);
11-
if (!engineFile)
12-
{
13-
std::cerr << "Error opening TensorRT Engine file at : " << engine_path << std::endl;
14-
}
15-
16-
engineFile.seekg(0, engineFile.end);
17-
long int fsize = engineFile.tellg();
18-
engineFile.seekg(0, engineFile.beg);
19-
20-
std::vector<char> engineData(fsize);
21-
engineFile.read(engineData.data(), fsize);
22-
if (!engineFile)
23-
{
24-
std::cerr << "Error loading engine from: " << engine_path << std::endl;
25-
}
26-
27-
return engineData;
28-
}
29-
308
int main(int argc, const char* argv[]) {
319
if (argc < 2) {
3210
std::cerr
@@ -35,7 +13,6 @@ int main(int argc, const char* argv[]) {
3513
}
3614

3715
std::string trt_ts_module_path = argv[1];
38-
// auto engineData = loadEngine(engine_path);
3916

4017
torch::jit::Module trt_ts_mod;
4118
try {

0 commit comments

Comments
 (0)