Skip to content

Commit bc88123

Browse files
peri044narendasan
authored andcommitted
chore: Minor fix
Signed-off-by: Dheeraj Peri <[email protected]>
1 parent a5aff8b commit bc88123

File tree

2 files changed

+5
-16
lines changed

2 files changed

+5
-16
lines changed

core/plugins/impl/interpolate_plugin.h

Lines changed: 0 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -30,16 +30,6 @@ class InterpolatePlugin : public nvinfer1::IPluginV2DynamicExt {
3030
bool align_corners_;
3131
bool use_scales_;
3232

33-
protected:
34-
// To prevent compiler warnings
35-
// using nvinfer1::IPluginV2DynamicExt::canBroadcastInputAcrossBatch;
36-
// using nvinfer1::IPluginV2DynamicExt::configurePlugin;
37-
// using nvinfer1::IPluginV2DynamicExt::enqueue;
38-
// using nvinfer1::IPluginV2DynamicExt::getOutputDimensions;
39-
// using nvinfer1::IPluginV2DynamicExt::getWorkspaceSize;
40-
// using nvinfer1::IPluginV2DynamicExt::isOutputBroadcastAcrossBatch;
41-
// using nvinfer1::IPluginV2DynamicExt::supportsFormat;
42-
4333
public:
4434
InterpolatePlugin(
4535
std::vector<int64_t> in_shape,

cpp/trtorchexec/main.cpp

Lines changed: 5 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -57,13 +57,12 @@ int main(int argc, const char* argv[]) {
5757

5858
auto compile_spec = trtorch::CompileSpec(dims);
5959
compile_spec.workspace_size = 1 << 24;
60-
compile_spec.sparse_weights = true;
6160

62-
// std::cout << "Checking operator support" << std::endl;
63-
// if (!trtorch::CheckMethodOperatorSupport(mod, "forward")) {
64-
// std::cerr << "Method is not currently supported by TRTorch" << std::endl;
65-
// return -1;
66-
// }
61+
std::cout << "Checking operator support" << std::endl;
62+
if (!trtorch::CheckMethodOperatorSupport(mod, "forward")) {
63+
std::cerr << "Method is not currently supported by TRTorch" << std::endl;
64+
return -1;
65+
}
6766

6867
std::cout << "Compiling graph to save as TRT engine (/tmp/engine_converted_from_jit.trt)" << std::endl;
6968
auto engine = trtorch::ConvertGraphToTRTEngine(mod, "forward", compile_spec);

0 commit comments

Comments
 (0)