From 0be7a0e0d9eb12a148ae51057634edbe1831ca61 Mon Sep 17 00:00:00 2001
From: Michael Abbott <me@escbook>
Date: Tue, 17 Nov 2020 08:59:39 +0100
Subject: [PATCH] one more pointer

---
 src/nnlib.jl | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/src/nnlib.jl b/src/nnlib.jl
index 1b37fe2b1d..1cf65bdd26 100644
--- a/src/nnlib.jl
+++ b/src/nnlib.jl
@@ -26,3 +26,6 @@ end
 # 1st argument is produced by NNlib.storage_type(A)
 NNlib._batched_gemm!(::Type{<:CuArray}, transA::Char, transB::Char, α::Number, A, B, β::Number, C) =
      CUBLAS.gemm_strided_batched!(transA, transB, α, A, B, β, C)
+
+Base.unsafe_convert(::Type{CuPtr{T}}, A::NNlib.BatchedAdjOrTrans{T}) where {T} =
+    Base.unsafe_convert(CuPtr{T}, parent(A))