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

Functions

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

Variables

 flags = initConfigFlags()
 NumOffloadTrheads
 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 kwargs.setdefault("InferenceTool", acc.popToolsAndMerge(
13 TritonToolCfg(flags, "MNIST_testModel", "localhost", name="EvaluateModelTritonTool")
14 ))
15
16
17 input_data = "dev/MLTest/2020-03-31/t10k-images-idx3-ubyte"
18 kwargs.setdefault("BatchSize", 2)
19 kwargs.setdefault("InputDataPixel", input_data)
20 kwargs.setdefault("OutputLevel", Constants.DEBUG)
21 acc.addEventAlgo(CompFactory.AthInfer.ExampleAsyncMLInferenceWithTriton(name, **kwargs))
22
23 return acc
24

Variable Documentation

◆ acc

AthExTritonAsync_test.acc = MainServicesCfg(flags)

Definition at line 37 of file AthExTritonAsync_test.py.

◆ flags

AthExTritonAsync_test.flags = initConfigFlags()

Definition at line 32 of file AthExTritonAsync_test.py.

◆ FPE

AthExTritonAsync_test.FPE

Definition at line 34 of file AthExTritonAsync_test.py.

◆ NumOffloadTrheads

AthExTritonAsync_test.NumOffloadTrheads

Definition at line 33 of file AthExTritonAsync_test.py.

◆ summariseProps

AthExTritonAsync_test.summariseProps

Definition at line 39 of file AthExTritonAsync_test.py.

◆ True

AthExTritonAsync_test.True

Definition at line 39 of file AthExTritonAsync_test.py.

◆ withDetails

AthExTritonAsync_test.withDetails

Definition at line 39 of file AthExTritonAsync_test.py.