Skip to content

Commit

Permalink
Add the DDFastShowerML package (#572)
Browse files Browse the repository at this point in the history
  • Loading branch information
jmcarcell authored Mar 19, 2024
1 parent 162a273 commit 87c7685
Show file tree
Hide file tree
Showing 4 changed files with 37 additions and 0 deletions.
2 changes: 2 additions & 0 deletions environments/key4hep-nightly/packages.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -15,6 +15,8 @@ packages:
variants: build_type=RelWithDebInfo
dd4hep:
variants: build_type=RelWithDebInfo
ddfastshowerml:
variants: build_type=RelWithDebInfo
ddkaltest:
variants: build_type=RelWithDebInfo
ddmarlinpandora:
Expand Down
33 changes: 33 additions & 0 deletions packages/ddfastshowerml/package.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,33 @@
# Copyright 2013-2020 Lawrence Livermore National Security, LLC and other
# Spack Project Developers. See the top-level COPYRIGHT file for details.
#
# SPDX-License-Identifier: (Apache-2.0 OR MIT)

from spack.pkg.k4.key4hep_stack import Ilcsoftpackage


class Ddfastshowerml(CMakePackage, Ilcsoftpackage):
"""Package with utilities and plugins that allow to run fast simulation in Geant4 from ML inference within ddsim (DDG4)"""

homepage = "https://gitlab.desy.de/ilcsoft/ddfastshowerml"
git = "https://gitlab.desy.de/ilcsoft/ddfastshowerml.git"
url = "https://gitlab.desy.de/ilcsoft/ddfastshowerml"

maintainers = ["jmcarcell"]

version("main", branch="main")

variant("inference", values=("onnxruntime", "torch", "both"), default="both")

depends_on("py-onnxruntime", when="inference=onnxruntime")
depends_on("py-onnxruntime", when="inference=both")
depends_on("py-torch", when="inference=torch")
depends_on("py-torch", when="inference=both")
depends_on("dd4hep")

def cmake_args(self):
args = []
args.append(
f"-DCMAKE_CXX_STANDARD={self.spec['root'].variants['cxxstd'].value}"
)
return args
1 change: 1 addition & 0 deletions packages/ilcsoft/package.py
Original file line number Diff line number Diff line change
Expand Up @@ -46,6 +46,7 @@ class Ilcsoft(BundlePackage, Key4hepPackage):
depends_on("clicperformance")
depends_on("clupatra")
depends_on("ced")
depends_on("ddfastshowerml")
depends_on("ddkaltest")
depends_on("ddmarlinpandora")
depends_on("fcalclusterer")
Expand Down
1 change: 1 addition & 0 deletions scripts/fetch_nightly_versions.py
Original file line number Diff line number Diff line change
Expand Up @@ -91,6 +91,7 @@ def get_latest_commit(
("clicperformance", "ilcsoft/clicperformance"),
("conformaltracking", "ilcsoft/conformaltracking"),
("dd4hep", "aidasoft/dd4hep"),
("ddfastshowerml", "ilcsoft/ddfastshowerml"),
("ddkaltest", "ilcsoft/ddkaltest"),
("ddmarlinpandora", "ilcsoft/ddmarlinpandora"),
("delphes", "delphes/delphes"),
Expand Down

0 comments on commit 87c7685

Please sign in to comment.