From 825532783c2e1df7e3868f71f821bbe31f2d0f7f Mon Sep 17 00:00:00 2001
From: Takafumi Arakaki <aka.tkf@gmail.com>
Date: Wed, 29 May 2019 18:22:36 -0700
Subject: [PATCH 1/2] Test Tracker.jacobian on GPU

---
 test/cuda/cuda.jl | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/test/cuda/cuda.jl b/test/cuda/cuda.jl
index 96d04c284b..1ed8345285 100644
--- a/test/cuda/cuda.jl
+++ b/test/cuda/cuda.jl
@@ -48,6 +48,13 @@ end
   @test y[3,:] isa CuArray
 end
 
+@testset "Jacobian on GPU" begin
+  # https://github.com/FluxML/Tracker.jl/pull/33
+  @test collect(jacobian(identity, gpu([0.0, 0.0]))) == [1 0; 0 1]
+  @test collect(gradient(x -> sum(jacobian(y -> y .^ 2, x) .^ 2),
+                         gpu([1.0, 2.0, 3.0]))) == [8.0, 16.0, 24.0]
+end
+
 if CuArrays.libcudnn != nothing
     @info "Testing Flux/CUDNN"
     include("cudnn.jl")

From 8c6ef2048c842557feec3c0b012b0ec1299f615f Mon Sep 17 00:00:00 2001
From: Takafumi Arakaki <aka.tkf@gmail.com>
Date: Wed, 29 May 2019 18:25:57 -0700
Subject: [PATCH 2/2] [DON'T MERGE] Use Tracker#remotes/cache/pull/33/head

---
 Manifest.toml | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/Manifest.toml b/Manifest.toml
index 8f2f0fadf8..13dcbbbd11 100644
--- a/Manifest.toml
+++ b/Manifest.toml
@@ -272,9 +272,11 @@ version = "0.5.3"
 
 [[Tracker]]
 deps = ["Adapt", "DiffRules", "ForwardDiff", "LinearAlgebra", "MacroTools", "NNlib", "NaNMath", "Printf", "Random", "Requires", "SpecialFunctions", "Statistics", "Test"]
-git-tree-sha1 = "0bec1b68c63a0e8a58d3944261cbf4cc9577c8a1"
+git-tree-sha1 = "2e516162af7f1e208b62ef3da3af0d094a44aab1"
+repo-rev = "9c2c05ec6d785996d83f7f59e3209461fcc36c01"
+repo-url = "https://github.com/FluxML/Tracker.jl.git"
 uuid = "9f7883ad-71c0-57eb-9f7f-b5c9e6d3789c"
-version = "0.2.0"
+version = "0.2.2"
 
 [[TranscodingStreams]]
 deps = ["Random", "Test"]