Apply suggestions from code review

Author: giordano

ext/ReactantCUDAExt.jl

@@ -352,7 +352,7 @@ function ka_with_reactant(ndrange, workgroupsize, obj, args...)
 
     # figure out the optimal workgroupsize automatically
     if KA.workgroupsize(obj) <: KA.DynamicSize && workgroupsize === nothing
-        if !Reactant.Compiler.PartitionKA[] || raising() || backend() in ("cpu", "tpu")
+        if !Reactant.Compiler.PartitionKA[] || raising() || Reactant.Compiler.backend() in ("cpu", "tpu")
             threads = prod(ndrange)
         else
             config = CUDA.launch_configuration(kernel.fun; max_threads=prod(ndrange))

src/Compiler.jl

@@ -720,7 +720,7 @@ function backend(; throw_error::Bool=true)
     if !(haskey(task_local_storage(), key) && !Base.isempty(task_local_storage(key)))
         throw_error && error("No Reactant backend context")
     end
-    return last(task_local_storage(key)::Vector{Bool})
+    return last(task_local_storage(key)::Vector{String})
 end
 
 function backend!(f, backend::String)