remove cufunc hack

Author: MikeInnes

src/cubroadcast.jl

@@ -1,39 +1,2 @@
-# This should probably be its own package
-
-_cufunc(f,x) = f
-_cufunc(f,x,xs...) = _cufunc(f, xs...)
-
-using MacroTools
-
-isbcastop(x) = isexpr(x, :call) && x.args[1] in :[.*,./,.+,.-].args
-
-broadcast_inputs(ex) =
-  ex isa Symbol ? [ex] :
-  @capture(ex, f_.(args__)) ? vcat(broadcast_inputs.(args)...) :
-  isbcastop(ex) ? vcat(broadcast_inputs.(ex.args[2:end])...) :
-  []
-
-cudata(x) = x
-
-"""
-Replace function calls with cuda intrinsics when applicable.
-
-@see CuArrays.cufunc
-"""
-macro fix(ex)
-  fs = []
-  ex = MacroTools.prewalk(ex) do ex
-    @capture(ex, f_.(args__)) || return ex
-    # May not work in cases like `x .+ log(1.0)` but w/e
-    xs = broadcast_inputs(ex)
-    f_ = gensym()
-    push!(fs, :($f_ = $_cufunc($f, $cudata.(($(xs...),))...)))
-    :($f_.($(args...)))
-  end
-
-  quote
-    let $(fs...)
-      $ex
-    end
-  end |> esc
-end
+# Kept only for backwards compatibility
+macro @fix(ex) esc(ex) end

test/cubroadcast.jl

@@ -4,9 +4,6 @@ using NNlib, Test
 
 include("activation.jl")
 include("conv.jl")
-# if Base.find_in_path("CuArrays") ≠ nothing
-#   include("cubroadcast.jl")
-# end
 
 xs = [-100_000, -100_000.]
 @test softmax(xs) ≈ [0.5, 0.5]