v0.1.1 released

eco-hydro · May 7, 2024 · 64039d6 · 64039d6
1 parent e407ace
commit 64039d6
Show file tree

Hide file tree

Showing 4 changed files with 24 additions and 2 deletions.
diff --git a/codecov.yml b/codecov.yml
@@ -4,3 +4,4 @@ ignore:
   - "src/*/temp"
   - "ext/*/test"
   - "src/*/backup"
+  - "src/Smooth/Whittaker/WHIT_GPU.jl"
diff --git a/src/Smooth/Whittaker/WHIT_GPU.jl b/src/Smooth/Whittaker/WHIT_GPU.jl
@@ -0,0 +1,21 @@
+using CUDA
+
+function WHIT_GPU(y::AbstractVector, w::AbstractVector, x::AbstractVector;
+  lambda=2.0, d=2, ignore...)
+  # Move data to GPU
+  y = CUDA.CuArray(y)
+  w = CUDA.CuArray(w)
+  x = CUDA.CuArray(x)
+
+  n = length(y)
+  D = ddmat(x, d)
+
+  W = SparseArrays.spdiagm(w)
+  A = W + lambda * D' * D
+
+  # L = cholesky(A).L # Matrix
+  L = cholesky(A, perm=1:n).L # sparse
+  z = L' \ (L \ (w .* y))
+  z
+end
+# export WHIT, speye, diff, ddmat
diff --git a/test/test-curvefit.jl b/test/test-curvefit.jl
@@ -1,6 +1,6 @@
 using Test
 using JLD2, UnPack, DataFrames
-@unpack y, w, t, ylu, dt, nptperyear = load("../data/phenofit-CA_NS6.jld2")
+@unpack y, w, t, ylu, dt, nptperyear = load("../data/phenofit_CA-NS6.jld2")
 
 @testset "curvefits" begin
   @test_nowarn begin

diff --git a/test/test-plot_input.jl b/test/test-plot_input.jl
@@ -4,7 +4,7 @@ using JLD2, UnPack
 
 
 @testset "plot_input" begin
-  @unpack y, w, t, QC_flag, ylu, dt, nptperyear = load("../data/phenofit-CA_NS6.jld2")
+  @unpack y, w, t, QC_flag, ylu, dt, nptperyear = load("../data/phenofit_CA-NS6.jld2")
   @test_nowarn p = plot_input(t, y, QC_flag)
 end