File tree Expand file tree Collapse file tree 2 files changed +5
-16
lines changed Expand file tree Collapse file tree 2 files changed +5
-16
lines changed Original file line number Diff line number Diff line change @@ -30,16 +30,6 @@ class InterpolatePlugin : public nvinfer1::IPluginV2DynamicExt {
30
30
bool align_corners_;
31
31
bool use_scales_;
32
32
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
-
43
33
public:
44
34
InterpolatePlugin (
45
35
std::vector<int64_t > in_shape,
Original file line number Diff line number Diff line change @@ -57,13 +57,12 @@ int main(int argc, const char* argv[]) {
57
57
58
58
auto compile_spec = trtorch::CompileSpec (dims);
59
59
compile_spec.workspace_size = 1 << 24 ;
60
- compile_spec.sparse_weights = true ;
61
60
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
+ }
67
66
68
67
std::cout << " Compiling graph to save as TRT engine (/tmp/engine_converted_from_jit.trt)" << std::endl;
69
68
auto engine = trtorch::ConvertGraphToTRTEngine (mod, " forward" , compile_spec);
You can’t perform that action at this time.
0 commit comments