update notebook - autodiff

GiggleLiu · May 7, 2021 · aba8b80 · aba8b80
1 parent 7e67362
commit aba8b80
Show file tree

Hide file tree

Showing 2 changed files with 1,100 additions and 23 deletions.
diff --git a/examples/batched_tr.jl b/examples/batched_tr.jl
@@ -1,5 +1,5 @@
 using NiLang, NiLang.AD
-using KernelAbstractions, CUDA
+using KernelAbstractions, CUDA, CUDAKernels
 
 @i @kernel function kernel_f(A, B::AbstractVector{TB}) where TB
     # turng off reversibility check, since GPU can not handle errors