Check if TRT version is greater than or equal to major.minor
(major: int, minor: int = 0)
| 127 | |
| 128 | |
| 129 | def trt_gte(major: int, minor: int = 0): |
| 130 | """ |
| 131 | Check if TRT version is greater than or equal to major.minor |
| 132 | """ |
| 133 | trt_ver = version.parse(trt_version()) |
| 134 | return trt_ver.major >= major and trt_ver.minor >= minor |
| 135 | |
| 136 | |
| 137 | def torch_version(): |
no test coverage detected