Skip to content

Commit

Permalink
nvrtc cleanup causes intermittent segfault, so remove libtorch testing (
Browse files Browse the repository at this point in the history
  • Loading branch information
deadeyegoodwin authored Aug 29, 2019
1 parent fc57c22 commit 8785507
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions qa/L0_sequence_batcher/sequence_batcher_test.py
Original file line number Diff line number Diff line change
Expand Up @@ -45,11 +45,11 @@
_model_instances = int(os.environ['MODEL_INSTANCES'])

if _no_batching:
_trials = ("savedmodel_nobatch", "graphdef_nobatch", "netdef_nobatch", "plan_nobatch", "onnx_nobatch", "libtorch_nobatch")
_trials = ("savedmodel_nobatch", "graphdef_nobatch", "netdef_nobatch", "plan_nobatch", "onnx_nobatch")
elif os.environ['BATCHER_TYPE'] == "VARIABLE":
_trials = ("savedmodel", "graphdef", "netdef", "onnx", "libtorch")
_trials = ("savedmodel", "graphdef", "netdef", "onnx")
else:
_trials = ("custom", "savedmodel", "graphdef", "netdef", "plan", "onnx", "libtorch")
_trials = ("custom", "savedmodel", "graphdef", "netdef", "plan", "onnx")
# Add ensemble to the _trials
ENSEMBLE_PREFIXES = ["simple_", "sequence_", "fan_"]
res = []
Expand Down

0 comments on commit 8785507

Please sign in to comment.