Skip to content

Commit

Permalink
Start
Browse files Browse the repository at this point in the history
  • Loading branch information
huiyuxie committed Jan 25, 2025
1 parent f29d3d4 commit 0accbc4
Showing 1 changed file with 5 additions and 4 deletions.
9 changes: 5 additions & 4 deletions src/solvers/basis_lobatto_legendre.jl
Original file line number Diff line number Diff line change
Expand Up @@ -44,8 +44,9 @@ function LobattoLegendreBasisGPU(polydeg::Integer, RealT = Float64) # how about
weights = CuArray{RealT}(weights_)
inverse_weights = CuArray{RealT}(inverse_weights_)

inverse_vandermonde_legendre = CuArray{RealT}(inverse_vandermonde_legendre_)
# boundary_interpolation = CuArray(boundary_interpolation_) # avoid scalar indexing
# Avoid scalar indexing
# inverse_vandermonde_legendre = CuArray{RealT}(inverse_vandermonde_legendre_)
# boundary_interpolation = CuArray(boundary_interpolation_)

derivative_matrix = CuArray{RealT}(derivative_matrix_)
derivative_split = CuArray{RealT}(derivative_split_)
Expand All @@ -54,11 +55,11 @@ function LobattoLegendreBasisGPU(polydeg::Integer, RealT = Float64) # how about

# TODO: Implement a custom struct for finer control over data types
return LobattoLegendreBasis{RealT, nnodes_, typeof(nodes),
typeof(inverse_vandermonde_legendre),
typeof(inverse_vandermonde_legendre_),
typeof(boundary_interpolation_),
typeof(derivative_matrix)}(nodes_, weights,
inverse_weights,
inverse_vandermonde_legendre,
inverse_vandermonde_legendre_,
boundary_interpolation_,
derivative_matrix,
derivative_split,
Expand Down

0 comments on commit 0accbc4

Please sign in to comment.