Some fixes (#114)

* Define `eachcol` in optional AD backends * Disable tests on Windows for now * Disable tests on 32bit as well
TuringLang · Jun 17, 2020 · 5216dd6 · 5216dd6
1 parent a72665d
commit 5216dd6
Show file tree

Hide file tree

Showing 6 changed files with 7 additions and 4 deletions.
diff --git a/.github/workflows/ForwardDiff_Tracker.yml b/.github/workflows/ForwardDiff_Tracker.yml
@@ -18,7 +18,6 @@ jobs:
         os:
           - ubuntu-latest
           - macOS-latest
-          - windows-latest
         arch:
           - x64
     steps:

diff --git a/.github/workflows/ReverseDiff.yml b/.github/workflows/ReverseDiff.yml
@@ -18,7 +18,6 @@ jobs:
         os:
           - ubuntu-latest
           - macOS-latest
-          - windows-latest
         arch:
           - x64
     steps:

diff --git a/.github/workflows/Zygote.yml b/.github/workflows/Zygote.yml
@@ -17,7 +17,6 @@ jobs:
         os:
           - ubuntu-latest
           - macOS-latest
-          - windows-latest
         arch:
           - x64
     steps:

diff --git a/src/compat/reversediff.jl b/src/compat/reversediff.jl
@@ -9,6 +9,8 @@ using ..Bijectors: Log, SimplexBijector, maphcat, simplex_link_jacobian,
     ReverseDiffAD, Inverse
 import ..Bijectors: _eps, logabsdetjac, _logabsdetjac_scale, _simplex_bijector, 
     _simplex_inv_bijector, replace_diag, jacobian, getpd, lower
+
+using Compat: eachcol
 using Distributions: LocationScale
 
 # AD implementations
@@ -178,4 +180,4 @@ lower(A::TrackedMatrix) = track(lower, A)
     return lower(Ad), Δ -> (lower(Δ),)
 end
 
-end
+end
diff --git a/src/compat/tracker.jl b/src/compat/tracker.jl
@@ -8,6 +8,8 @@ using .Tracker: Tracker,
                 track,
                 data,
                 param
+
+using Compat: eachcol
 using LinearAlgebra
 
 maporbroadcast(f, x::TrackedArray...) = f.(x...)

diff --git a/src/compat/zygote.jl b/src/compat/zygote.jl
@@ -1,5 +1,7 @@
 using .Zygote: Zygote, @adjoint, @nograd, pullback
 
+using Compat: eachcol
+
 @adjoint istraining() = true, _ -> nothing
 @nograd Bijectors._debug