ATLAS Offline Software
Loading...
Searching...
No Matches
AthExTritonAsync_test Namespace Reference

Functions

 AthExTritonCfg (flags, name="AthExTritonExample", **kwargs)

Variables

 flags = initConfigFlags()
 NumOffloadThreads
 FPE
 acc = MainServicesCfg(flags)
 withDetails
 True
 summariseProps

Function Documentation

◆ AthExTritonCfg()

AthExTritonAsync_test.AthExTritonCfg ( flags,
name = "AthExTritonExample",
** kwargs )

Definition at line 8 of file AthExTritonAsync_test.py.

8def AthExTritonCfg(flags, name="AthExTritonExample", **kwargs):
9 acc = ComponentAccumulator()
10
11 from AthTritonComps.TritonToolConfig import TritonToolCfg
12 try:
13 kwargs.setdefault("InferenceTool", acc.popToolsAndMerge(
14 TritonToolCfg(flags, "MNIST_testModel", "localhost",
15 name="EvaluateModelTritonTool")
16 ))
17 except RuntimeError as e:
18 import sys
19 from AthenaCommon.Logging import log as msg
20 msg.warning(e)
21 sys.exit(2) # indicate test is skipped, not failed
22
23 input_data = "dev/MLTest/2020-03-31/t10k-images-idx3-ubyte"
24 kwargs.setdefault("BatchSize", 2)
25 kwargs.setdefault("InputDataPixel", input_data)
26 kwargs.setdefault("OutputLevel", Constants.DEBUG)
27 acc.addEventAlgo(
28 CompFactory.AthInfer.ExampleAsyncMLInferenceWithTriton(name, **kwargs))
29
30 return acc
31
32

Variable Documentation

◆ acc

AthExTritonAsync_test.acc = MainServicesCfg(flags)

Definition at line 45 of file AthExTritonAsync_test.py.

◆ flags

AthExTritonAsync_test.flags = initConfigFlags()

Definition at line 40 of file AthExTritonAsync_test.py.

◆ FPE

AthExTritonAsync_test.FPE

Definition at line 42 of file AthExTritonAsync_test.py.

◆ NumOffloadThreads

AthExTritonAsync_test.NumOffloadThreads

Definition at line 41 of file AthExTritonAsync_test.py.

◆ summariseProps

AthExTritonAsync_test.summariseProps

Definition at line 47 of file AthExTritonAsync_test.py.

◆ True

AthExTritonAsync_test.True

Definition at line 47 of file AthExTritonAsync_test.py.

◆ withDetails

AthExTritonAsync_test.withDetails

Definition at line 47 of file AthExTritonAsync_test.py.