Skip to content

DNNROI GPU Config #827

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
wants to merge 7 commits into
base: develop
Choose a base branch
from
Open

DNNROI GPU Config #827

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
7 commits
Select commit Hold shift + click to select a range
e3ec6f9
Update stage0_icarus_defs.fcl
nathanielerowe May 20, 2025
36fcfae
Create wcls-decode-to-sig-twofaced-dnnroi-gpu.jsonnet
nathanielerowe May 20, 2025
8ce554f
Delete icaruscode/TPC/ICARUSWireCell/icarus/wcls-decode-to-sig-twofac…
nathanielerowe May 25, 2025
393c207
Update wcls-decode-to-sig-twofaced-dnnroi.jsonnet
nathanielerowe May 25, 2025
a360cec
Update stage0_run2_wcdnn_icarus_overlay.fcl
nathanielerowe May 25, 2025
14cdfa4
Create stage0_run2_wcdnn_gpu_icarus.fcl
nathanielerowe May 25, 2025
08a47ad
Update stage0_icarus_defs.fcl
nathanielerowe May 25, 2025
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
13 changes: 13 additions & 0 deletions fcl/reco/Stage0/overlay/stage0_run2_wcdnn_gpu_icarus.fcl
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -0,0 +1,13 @@
#include "stage0_run2_wc_icarus.fcl"

#include "enable_dnn_sp.fcl"

physics.producers.decon2droiEE.wcls_main.params.wc_device: "gpu"
physics.producers.decon2droiEW.wcls_main.params.wc_device: "gpu"
physics.producers.decon2droiWE.wcls_main.params.wc_device: "gpu"
physics.producers.decon2droiWW.wcls_main.params.wc_device: "gpu"

physics.producers.decon2droiEE.wcls_main.plugins: [@sequence::icarus_stage0_producers.decon2droiEE.wcls_main.plugins, "WireCellCuda"]
physics.producers.decon2droiEW.wcls_main.plugins: [@sequence::icarus_stage0_producers.decon2droiEW.wcls_main.plugins, "WireCellCuda"]
physics.producers.decon2droiWE.wcls_main.plugins: [@sequence::icarus_stage0_producers.decon2droiWE.wcls_main.plugins, "WireCellCuda"]
physics.producers.decon2droiWW.wcls_main.plugins: [@sequence::icarus_stage0_producers.decon2droiWW.wcls_main.plugins, "WireCellCuda"]
8 changes: 7 additions & 1 deletion fcl/reco/Stage0/overlay/stage0_run2_wcdnn_icarus_overlay.fcl
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -1,3 +1,9 @@
#include "stage0_run2_wcdnn_icarus_mc.fcl"
#

#include "enable_overlay_sp.fcl"

# for gpus, use stage0_run2_wcdnn_gpu_icarus.fcl instead
physics.producers.decon2droiEE.wcls_main.params.wc_device: "cpu"
physics.producers.decon2droiEW.wcls_main.params.wc_device: "cpu"
physics.producers.decon2droiWE.wcls_main.params.wc_device: "cpu"
physics.producers.decon2droiWW.wcls_main.params.wc_device: "cpu"
10 changes: 7 additions & 3 deletions icaruscode/TPC/ICARUSWireCell/icarus/wcls-decode-to-sig-twofaced-dnnroi.jsonnet
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -19,6 +19,7 @@
// -V signal_output_form=sparse \\
// -J cfg cfg/pgrapher/experiment/uboone/wcls-nf-sp.jsonnet

local wc_device = std.extVar('wc_device');

local epoch = std.extVar('epoch'); // eg "dynamic", "after", "before", "perfect"
local reality = std.extVar('reality');
Expand Down Expand Up @@ -87,7 +88,10 @@ local params = base {

// local tools_maker = import 'pgrapher/common/tools.jsonnet';
local tools_maker = import 'pgrapher/experiment/icarus/icarus_tools.jsonnet';
local tools = tools_maker(params);
// local tools = tools_maker(params);
local default_tools = tools_maker(params);
local tools = if wc_device == 'gpu' then std.mergePatch(default_tools,
{dft: {type: "TorchDFT", data: {device: wc_device}}}) else default_tools;


local wcls_maker = import 'pgrapher/ui/wcls/nodes.jsonnet';
Expand Down Expand Up @@ -271,7 +275,7 @@ local ts_u = {
name: "dnnroi_u",
data: {
model: "NNs/plane0.ts",
device: "cpu",
device: wc_device,
concurrency: 1,
},
};
Expand All @@ -281,7 +285,7 @@ local ts_v = {
name: "dnnroi_v",
data: {
model: "NNs/plane1.ts",
device: "cpu",
device: wc_device,
concurrency: 1,
},
};
Expand Down