@@ -10,27 +10,26 @@ namespace {
10
10
11
11
void RegisterTRTCompileSpec () {
12
12
static auto TRTORCH_UNUSED TRTInputRangeTSRegistration =
13
- torch::class_<trtorch::pyapi::InputRange>(" tensorrt" , " InputRange" ).def (torch::init<>());
13
+ torch::class_<trtorch::pyapi::InputRange>(" tensorrt" , " InputRange" ).def (torch::init<>());
14
14
15
15
ADD_FIELD_GET_SET_REGISTRATION (TRTInputRangeTSRegistration, trtorch::pyapi::InputRange, min);
16
16
ADD_FIELD_GET_SET_REGISTRATION (TRTInputRangeTSRegistration, trtorch::pyapi::InputRange, opt);
17
17
ADD_FIELD_GET_SET_REGISTRATION (TRTInputRangeTSRegistration, trtorch::pyapi::InputRange, max);
18
18
19
19
static auto TRTORCH_UNUSED TRTDeviceTSRegistration =
20
- torch::class_<trtorch::pyapi::Device>(" tensorrt" , " Device" ).def (torch::init<>());
20
+ torch::class_<trtorch::pyapi::Device>(" tensorrt" , " Device" ).def (torch::init<>());
21
21
22
22
ADD_FIELD_GET_SET_REGISTRATION (TRTDeviceTSRegistration, trtorch::pyapi::Device, device_type);
23
23
ADD_FIELD_GET_SET_REGISTRATION (TRTDeviceTSRegistration, trtorch::pyapi::Device, gpu_id);
24
24
ADD_FIELD_GET_SET_REGISTRATION (TRTDeviceTSRegistration, trtorch::pyapi::Device, dla_core);
25
25
ADD_FIELD_GET_SET_REGISTRATION (TRTDeviceTSRegistration, trtorch::pyapi::Device, allow_gpu_fallback);
26
26
27
-
28
27
static auto TRTORCH_UNUSED TRTCompileSpecTSRegistration =
29
- torch::class_<trtorch::pyapi::CompileSpec>(" tensorrt" , " CompileSpec" )
30
- .def (torch::init<>())
31
- .def (" append_input_range" , &trtorch::pyapi::CompileSpec::appendInputRange)
32
- .def (" set_device" , &trtorch::pyapi::CompileSpec::setDeviceIntrusive)
33
- .def (" __str__" , &trtorch::pyapi::CompileSpec::stringify);
28
+ torch::class_<trtorch::pyapi::CompileSpec>(" tensorrt" , " CompileSpec" )
29
+ .def (torch::init<>())
30
+ .def (" append_input_range" , &trtorch::pyapi::CompileSpec::appendInputRange)
31
+ .def (" set_device" , &trtorch::pyapi::CompileSpec::setDeviceIntrusive)
32
+ .def (" __str__" , &trtorch::pyapi::CompileSpec::stringify);
34
33
35
34
ADD_FIELD_GET_SET_REGISTRATION (TRTCompileSpecTSRegistration, trtorch::pyapi::CompileSpec, op_precision);
36
35
ADD_FIELD_GET_SET_REGISTRATION (TRTCompileSpecTSRegistration, trtorch::pyapi::CompileSpec, refit);
@@ -41,7 +40,6 @@ void RegisterTRTCompileSpec() {
41
40
ADD_FIELD_GET_SET_REGISTRATION (TRTCompileSpecTSRegistration, trtorch::pyapi::CompileSpec, num_avg_timing_iters);
42
41
ADD_FIELD_GET_SET_REGISTRATION (TRTCompileSpecTSRegistration, trtorch::pyapi::CompileSpec, workspace_size);
43
42
ADD_FIELD_GET_SET_REGISTRATION (TRTCompileSpecTSRegistration, trtorch::pyapi::CompileSpec, max_batch_size);
44
-
45
43
}
46
44
47
45
struct TRTTSRegistrations {
0 commit comments