Skip to content

Commit 8eec0c2

Browse files
fix an issue that the default run_card for GPU was the one for CPP and not the one for GPU
1 parent 9e254df commit 8eec0c2

1 file changed

Lines changed: 11 additions & 5 deletions

File tree

  • epochX/cudacpp/CODEGEN/PLUGIN/CUDACPP_SA_OUTPUT

epochX/cudacpp/CODEGEN/PLUGIN/CUDACPP_SA_OUTPUT/output.py

Lines changed: 11 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -37,6 +37,9 @@
3737
import madgraph.iolibs.files as files
3838
import madgraph.various.misc as misc
3939

40+
from . import launch_plugin
41+
42+
4043
# AV - define the plugin's process exporter
4144
# (NB: this is the plugin's main class, enabled in the new_output dictionary in __init__.py)
4245
class PLUGIN_ProcessExporter(PLUGIN_export_cpp.ProcessExporterGPU):
@@ -151,11 +154,6 @@ class PLUGIN_ProcessExporter(PLUGIN_export_cpp.ProcessExporterGPU):
151154
###helas_exporter = None
152155
helas_exporter = model_handling.PLUGIN_GPUFOHelasCallWriter # this is one of the main fixes for issue #341!
153156

154-
# Default class for the run_card to use
155-
from . import launch_plugin
156-
run_card_class = launch_plugin.CPPRunCard
157-
158-
159157
# AV (default from OM's tutorial) - add a debug printout
160158
def __init__(self, *args, **kwargs):
161159
self.in_madevent_mode = False # see MR #747
@@ -300,6 +298,10 @@ class PLUGIN_ProcessExporter_MadEvent(PLUGIN_ProcessExporter):
300298
#------------------------------------------------------------------------------------
301299

302300
class SIMD_ProcessExporter(PLUGIN_ProcessExporter_MadEvent):
301+
302+
# Default class for the run_card to use
303+
run_card_class = launch_plugin.CPPRunCard
304+
303305
def change_output_args(args, cmd):
304306
""" """
305307
cmd._export_format = "madevent"
@@ -312,6 +314,10 @@ def change_output_args(args, cmd):
312314
#------------------------------------------------------------------------------------
313315

314316
class GPU_ProcessExporter(PLUGIN_ProcessExporter_MadEvent):
317+
318+
# Default class for the run_card to use
319+
run_card_class = launch_plugin.GPURunCard
320+
315321
def change_output_args(args, cmd):
316322
""" """
317323
cmd._export_format = "madevent"

0 commit comments

Comments
 (0)