move particle bank operations out of kernel.py

Author: ilhamv

mcdc/code_factory/adapt.py

@@ -4,7 +4,7 @@
 from numba.extending import intrinsic
 import numba
 import mcdc.object_.numba_types as type_
-import mcdc.transport.kernel as kernel
+import mcdc.transport.particle_bank as particle_bank_module
 
 
 if importlib.util.find_spec("harmonize") is None:
@@ -515,13 +515,13 @@ def thread(prog):
 
 @for_cpu()
 def add_active(particle, prog):
-    kernel.add_particle(particle, prog["bank_active"])
+    particle_bank_module.add_particle(particle, prog["bank_active"])
 
 
 @for_gpu()
 def add_active(P_reclike, prog):
     P = local_array(1, type_.particle)
-    kernel.recordlike_to_particle(P, P_reclike)
+    particle_bank_module.recordlike_to_particle(P, P_reclike)
     if SIMPLE_ASYNC:
         step_async(prog, P[0])
     else:
@@ -530,35 +530,35 @@ def add_active(P_reclike, prog):
 
 @for_cpu()
 def add_source(particle, prog):
-    kernel.add_particle(particle, prog["bank_source"])
+    particle_bank_module.add_particle(particle, prog["bank_source"])
 
 
 @for_gpu()
 def add_source(particle, prog):
     mcdc = mcdc_global(prog)
-    kernel.add_particle(particle, mcdc["bank_source"])
+    particle_bank_module.add_particle(particle, mcdc["bank_source"])
 
 
 @for_cpu()
 def add_census(particle, prog):
-    kernel.add_particle(particle, prog["bank_census"])
+    particle_bank_module.add_particle(particle, prog["bank_census"])
 
 
 @for_gpu()
 def add_census(particle, prog):
     mcdc = mcdc_global(prog)
-    kernel.add_particle(particle, mcdc["bank_census"])
+    particle_bank_module.add_particle(particle, mcdc["bank_census"])
 
 
 @for_cpu()
 def add_future(particle, prog):
-    kernel.add_particle(particle, prog["bank_future"])
+    particle_bank_module.add_particle(particle, prog["bank_future"])
 
 
 @for_gpu()
 def add_future(particle, prog):
     mcdc = mcdc_global(prog)
-    kernel.add_particle(particle, mcdc["bank_future"])
+    particle_bank_module.add_particle(particle, mcdc["bank_future"])
 
 
 @for_cpu()

mcdc/main.py

@@ -228,7 +228,7 @@ def preparation():
     # Adapt kernels
     import numba as nb
     import mcdc.config as config
-    import mcdc.transport.kernel as kernel  # TODO: why is this needed here?
+    import mcdc.transport.particle_bank as particle_bank_module # TODO: why is this needed here?
     import mcdc.transport.mpi as mpi
 
     settings.target_gpu = True if config.target == "gpu" else False