Commit 378366b
authored
[6034518] Remove return statement preventing remote auto tuning (NVIDIA#1361)
### What does this PR do?
Remove return statement from the code checking remote auto tuning config
arguments since that results in skipping adding the actual remote tuning
config to the trtexec cmd.
**Root cause**: The necessary flags do not get added to
`self._base_cmd.extend(trtexec_args)` when remote autotuning is enabled.
**Before fix**:
```
['trtexec', '--avgRuns=100', '--iterations=100', '--warmUp=50', '--stronglyTyped', \
'--saveEngine=engine.trt', '--timingCacheFile=trtexec_timing.cache', \
'--onnx=baseline.onnx']
```
**After fix**:
```
['trtexec', '--avgRuns=100', '--iterations=100', '--warmUp=50', '--stronglyTyped', \
'--saveEngine=engine.trt', '--timingCacheFile=trtexec_timing.cache', \
'--remoteAutoTuningConfig=$CONFIG', '--safe', '--skipInference', \
'--onnx=baseline.onnx']
```
Notice that the remote autotuning and related flags are now included in
the `trtexec` command.
**Related PR**: NVIDIA#1259
<!-- This is an auto-generated comment: release notes by coderabbit.ai
-->
## Summary by CodeRabbit
## Bug Fixes
* Fixed an issue where remote autotuning configuration arguments were
not being properly included in benchmark commands, ensuring all remote
autotuning settings are now correctly applied during execution.
<!-- end of auto-generated comment: release notes by coderabbit.ai -->
Signed-off-by: dmoodie <dmoodie@nvidia.com>1 parent a492fa9 commit 378366b
1 file changed
Lines changed: 0 additions & 1 deletion
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
220 | 220 | | |
221 | 221 | | |
222 | 222 | | |
223 | | - | |
224 | 223 | | |
225 | 224 | | |
226 | 225 | | |
| |||
0 commit comments