Change chunksize

SciML · Feb 13, 2025 · d494f72 · d494f72
1 parent a76959a
commit d494f72
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/test/core_odes.jl b/test/core_odes.jl
@@ -10,7 +10,7 @@ function foo(du, u, (A, tmp), t)
     nothing
 end
 #with defined chunk_size
-chunk_size = 5
+chunk_size = 9
 u0 = ones(5, 5)
 A = ones(5, 5)
 cache = DiffCache(zeros(5, 5), chunk_size)

diff --git a/test/gpu_all.jl b/test/gpu_all.jl
@@ -56,7 +56,7 @@ function foo(du, u, (A, tmp), t)
     nothing
 end
 #with specified chunk_size
-chunk_size = 10
+chunk_size = 9
 u0 = cu(rand(10, 10)) #example kept small for test purposes.
 A = cu(-randn(10, 10))
 cache = DiffCache(cu(zeros(10, 10)), chunk_size)