Run on GPU

.gitattributes

@@ -1,5 +1,3 @@
 input/AB->ABBBBBBBBB.txt filter=lfs diff=lfs merge=lfs -text
 input/AB->ABBBBBBB.txt filter=lfs diff=lfs merge=lfs -text
-*.zip filter=lfs diff=lfs merge=lfs
-*.gif filter=lfs diff=lfs merge=lfs
-*.jld2 filter=lfs diff=lfs merge=lfs
+*.zip filter=lfs diff=lfs merge=lfs -text

src/QEDprocesses_patch.jl

@@ -1,12 +1,25 @@
 # patch QEDprocesses
 # see issue https://github.com/QEDjl-project/QEDprocesses.jl/issues/77
 @inline function QEDprocesses.number_particles(
-    proc_def::QEDbase.AbstractProcessDefinition, ::Type{PS}
+    proc_def::QEDbase.AbstractProcessDefinition,
+    ::Type{PS},
 ) where {
-    DIR<:QEDbase.ParticleDirection,
-    PT<:QEDbase.AbstractParticleType,
-    EL<:AbstractFourMomentum,
-    PS<:ParticleStateful{DIR,PT,EL},
+    DIR <: QEDbase.ParticleDirection,
+    PT <: QEDbase.AbstractParticleType,
+    EL <: AbstractFourMomentum,
+    PS <: ParticleStateful{DIR, PT, EL},
 }
     return QEDprocesses.number_particles(proc_def, DIR(), PT())
 end
+
+@inline function QEDcore.ParticleStateful{DIR, SPECIES}(
+    mom::AbstractFourMomentum,
+) where {DIR <: ParticleDirection, SPECIES <: AbstractParticleType}
+    return ParticleStateful(DIR(), SPECIES(), mom)
+end
+
+@inline function QEDcore.ParticleStateful{DIR, SPECIES, EL}(
+    mom::EL,
+) where {DIR <: ParticleDirection, SPECIES <: AbstractParticleType, EL <: AbstractFourMomentum}
+    return ParticleStateful(DIR(), SPECIES(), mom)
+end

src/code_gen/function.jl

@@ -16,7 +16,9 @@ function get_compute_function(graph::DAG, instance, machine::Machine)
         "function compute_$(functionId)(data_input::$(input_type(instance))) $(initCaches); $(assignInputs); $code; return $resSym; end",
     )
 
-    return expr
+    func = eval(expr)
+
+    return func
 end
 
 """
@@ -33,22 +35,22 @@ function get_cuda_kernel(graph::DAG, instance, machine::Machine)
 
     functionId = to_var_name(UUIDs.uuid1(rng[1]))
     resSym = eval(gen_access_expr(entry_device(tape.machine), tape.outputSymbol))
-    expr = Meta.parse(
-        "function compute_$(functionId)(input_vector, output_vector, n::Int64)
-            id = (blockIdx().x - 1) * blockDim().x + threadIdx().x
-            if (id > n)
-                return
-            end
-            @inline data_input = input_vector[id]
-            $(initCaches)
-            $(assignInputs)
-            $code
-            @inline output_vector[id] = $resSym
-            return nothing
-        end"
-    )
+    expr = Meta.parse("function compute_$(functionId)(input_vector, output_vector, n::Int64)
+                          id = (blockIdx().x - 1) * blockDim().x + threadIdx().x
+                          if (id > n)
+                              return
+                          end
+                          @inline data_input = input_vector[id]
+                          $(initCaches)
+                          $(assignInputs)
+                          $code
+                          @inline output_vector[id] = $resSym
+                          return nothing
+                      end")
 
-    return expr
+    func = eval(expr)
+
+    return func
 end
 
 """