PhysicsTools/PyTorchAlpakaTest/test/BuildFile.xml - master - githubmirror/cmssw

PhysicsTools/PyTorchAlpakaTest/test/BuildFile.xml

7 строк571 B

Christine Zeh

Integration of PyTorch based portable (direct) ML inference with Alpaka-based heterogeneous core

21 ноя 2025, 14:01

21 ноя 2025, 14:01a7218f1

<test name="testPyTorchAlpakaTestSerialSync" command="cmsRun ${LOCALTOP}/src/PhysicsTools/PyTorchAlpakaTest/test/runPyTorchAlpakaTest.py -b serial_sync -ne 8 -ns 2 -nt 2"/>
<test name="testPyTorchAlpakaTestCudaAsync" command="cmsRun ${LOCALTOP}/src/PhysicsTools/PyTorchAlpakaTest/test/runPyTorchAlpakaTest.py -b cuda_async -ne 8 -ns 2 -nt 2">
  <use name="cuda"/>
</test>
<test name="testPyTorchAlpakaTestROCmAsync" command="cmsRun ${LOCALTOP}/src/PhysicsTools/PyTorchAlpakaTest/test/runPyTorchAlpakaTest.py -b rocm_async -ne 8 -ns 2 -nt 2">
  <use name="rocm"/>
</test>