From 7aa53f82da1ffb4fcdc4171bf2e0275035d0fc5e Mon Sep 17 00:00:00 2001 From: Anton Reinhard Date: Tue, 17 Sep 2024 17:34:51 +0200 Subject: [PATCH] Add context_module argument back --- ext/devices/cuda/function.jl | 4 +++- ext/devices/rocm/function.jl | 4 +++- 2 files changed, 6 insertions(+), 2 deletions(-) diff --git a/ext/devices/cuda/function.jl b/ext/devices/cuda/function.jl index ef7920a..1bf5a5d 100644 --- 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 index 6990ba1..2ecca3b 100644 --- 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)