Add context_module argument back (#34)

ComputableDAGs · Sep 17, 2024 · 19e1617 · 19e1617
1 parent 695bdb6
commit 19e1617
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/ext/devices/cuda/function.jl b/ext/devices/cuda/function.jl
@@ -1,4 +1,6 @@
-function ComputableDAGs.kernel(::Type{CUDAGPU}, graph::DAG, instance)
+function ComputableDAGs.kernel(
+    ::Type{CUDAGPU}, graph::DAG, instance, context_module::Module
+)
     machine = cpu_st()
     tape = ComputableDAGs.gen_tape(graph, instance, machine, context_module)
 

diff --git a/ext/devices/rocm/function.jl b/ext/devices/rocm/function.jl
@@ -1,4 +1,6 @@
-function ComputableDAGs.kernel(::Type{ROCmGPU}, graph::DAG, instance)
+function ComputableDAGs.kernel(
+    ::Type{ROCmGPU}, graph::DAG, instance, context_module::Module
+)
     machine = cpu_st()
     tape = ComputableDAGs.gen_tape(graph, instance, machine, context_module)