From 4f2ad553b67fd6d63c18018f7242ba8a4a67f422 Mon Sep 17 00:00:00 2001 From: Jael Gu Date: Fri, 3 Feb 2023 13:39:32 +0800 Subject: [PATCH] Update due to change of triton_client Signed-off-by: Jael Gu --- benchmark/qps_test.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/benchmark/qps_test.py b/benchmark/qps_test.py index 57a6c3e..367a5df 100644 --- a/benchmark/qps_test.py +++ b/benchmark/qps_test.py @@ -66,7 +66,7 @@ if args.num and args.pipe: if args.triton: client = triton_client.Client(url='localhost:8000') - out2 = client(data)[0][0][0] + out2 = client(data)[0][0] print('Triton: OK') if numpy.allclose(out1, out2, atol=args.atol):