Merge #551

551: @Nograd printing r=willtebbutt a=willtebbutt Resolves #548 . Bumps patch so that we can make this available immediately. Co-authored-by: willtebbutt <[email protected]>
FluxML · Mar 20, 2020 · 24f9c58 · 24f9c58 · willtebbutt · Mar 23, 2020
2 parents f16d856 + 047d8e6
commit 24f9c58
Show file tree

Hide file tree

Showing 3 changed files with 16 additions and 2 deletions.
diff --git a/Project.toml b/Project.toml
@@ -1,6 +1,6 @@
 name = "Zygote"
 uuid = "e88e6eb3-aa80-5325-afca-941959d7151f"
-version = "0.4.10"
+version = "0.4.11"
 
 [deps]
 ArrayLayouts = "4c555306-a7a7-4459-81d9-ec55ddd5c99a"

diff --git a/src/lib/base.jl b/src/lib/base.jl
@@ -1,4 +1,5 @@
-@nograd readline, Base.gc_num, Base.time_ns
+@nograd readline, Base.gc_num, Base.time_ns, Base.print, Base.println, Base.show,
+  Core.show, Core.print, Core.println
 
 # Gradient of AD stacks
 

diff --git a/test/gradcheck.jl b/test/gradcheck.jl
@@ -40,6 +40,19 @@ end
 
 Random.seed!(0)
 
+@testset "println, show, print" begin
+  function foo(x)
+    Base.show(x)
+    Base.print(x)
+    Base.println(x)
+    Core.show(x)
+    Core.print(x)
+    Core.println(x)
+    return x
+  end
+  @test gradtest(foo, [5.0])
+end
+
 @test gradient(//, 2, 3) === (1//3, -2//9)
 
 @testset "power" begin