Skip to content
Projects
Groups
Snippets
Help
Loading...
Help
Support
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
L
LGPU FPP
Project overview
Project overview
Details
Activity
Releases
Repository
Repository
Files
Commits
Branches
Tags
Contributors
Graph
Compare
Issues
0
Issues
0
List
Boards
Labels
Milestones
Merge Requests
0
Merge Requests
0
CI / CD
CI / CD
Pipelines
Jobs
Schedules
Analytics
Analytics
CI / CD
Repository
Value Stream
Wiki
Wiki
Snippets
Snippets
Members
Members
Collapse sidebar
Close sidebar
Activity
Graph
Create a new issue
Jobs
Commits
Issue Boards
Open sidebar
Fernando Pérez Panadero
LGPU FPP
Commits
40cc784d
Commit
40cc784d
authored
1 year ago
by
Fernando Pérez Panadero
Browse files
Options
Browse Files
Download
Email Patches
Plain Diff
sfcf slightly modified
parent
3bd8f080
Changes
3
Hide whitespace changes
Inline
Side-by-side
Showing
3 changed files
with
13 additions
and
15 deletions
+13
-15
src/LatticeGPU.jl
src/LatticeGPU.jl
+1
-1
src/Solvers/Solvers.jl
src/Solvers/Solvers.jl
+1
-1
src/Solvers/sfcf.jl
src/Solvers/sfcf.jl
+11
-13
No files found.
src/LatticeGPU.jl
View file @
40cc784d
...
@@ -64,6 +64,6 @@ include("Solvers/Solvers.jl")
...
@@ -64,6 +64,6 @@ include("Solvers/Solvers.jl")
using
.
Solvers
using
.
Solvers
export
CG!
export
CG!
export
propagator
,
bndpropagator
export
propagator
,
bndpropagator
export
fP
!
,
fA!
,
kV!
,
lV!
export
fP
,
fA
,
kV
,
lV
end
# module
end
# module
This diff is collapsed.
Click to expand it.
src/Solvers/Solvers.jl
View file @
40cc784d
...
@@ -26,7 +26,7 @@ include("Propagators.jl")
...
@@ -26,7 +26,7 @@ include("Propagators.jl")
export
propagator
,
bndpropagator
export
propagator
,
bndpropagator
include
(
"sfcf.jl"
)
include
(
"sfcf.jl"
)
export
fP
!
,
fA!
,
kV!
,
lV!
export
fP
,
fA
,
kV
,
lV
end
end
This diff is collapsed.
Click to expand it.
src/Solvers/sfcf.jl
View file @
40cc784d
function
fP
!
(
fp
,
dws
::
DiracWorkspace
,
prop
::
AbstractArray
,
lp
::
SpaceParm
)
function
fP
(
prop
::
AbstractArray
,
dws
::
DiracWorkspace
,
lp
::
SpaceParm
)
function
krnl_fP
!
(
sum
,
pro
)
function
krnl_fP
!
(
sum
,
pro
)
b
=
Int64
(
CUDA
.
threadIdx
()
.
x
)
b
=
Int64
(
CUDA
.
threadIdx
()
.
x
)
...
@@ -16,12 +16,10 @@ function fP!(fp, dws::DiracWorkspace, prop::AbstractArray,lp::SpaceParm)
...
@@ -16,12 +16,10 @@ function fP!(fp, dws::DiracWorkspace, prop::AbstractArray,lp::SpaceParm)
CUDA
.
@cuda
threads
=
lp
.
bsz
blocks
=
lp
.
rsz
krnl_fP!
(
dws
.
cs
,
prop
)
CUDA
.
@cuda
threads
=
lp
.
bsz
blocks
=
lp
.
rsz
krnl_fP!
(
dws
.
cs
,
prop
)
end
end
fp
+=
reshape
(
Array
(
CUDA
.
reduce
(
+
,
dws
.
cs
;
dims
=
(
1
,
2
,
3
))),
lp
.
iL
[
end
])
return
reshape
(
Array
(
CUDA
.
reduce
(
+
,
dws
.
cs
;
dims
=
(
1
,
2
,
3
))),
lp
.
iL
[
end
])
return
nothing
end
end
function
fA
!
(
fa
,
dws
::
DiracWorkspace
,
prop
::
AbstractArray
,
lp
::
SpaceParm
)
function
fA
(
prop
::
AbstractArray
,
dws
::
DiracWorkspace
,
lp
::
SpaceParm
)
function
krnl_fA
!
(
sum
,
pro
)
function
krnl_fA
!
(
sum
,
pro
)
b
=
Int64
(
CUDA
.
threadIdx
()
.
x
)
b
=
Int64
(
CUDA
.
threadIdx
()
.
x
)
...
@@ -35,11 +33,11 @@ function fA!(fa, dws::DiracWorkspace, prop::AbstractArray,lp::SpaceParm)
...
@@ -35,11 +33,11 @@ function fA!(fa, dws::DiracWorkspace, prop::AbstractArray,lp::SpaceParm)
CUDA
.
@sync
begin
CUDA
.
@sync
begin
CUDA
.
@cuda
threads
=
lp
.
bsz
blocks
=
lp
.
rsz
krnl_fA!
(
dws
.
cs
,
prop
)
CUDA
.
@cuda
threads
=
lp
.
bsz
blocks
=
lp
.
rsz
krnl_fA!
(
dws
.
cs
,
prop
)
end
end
fa
+=
reshape
(
Array
(
CUDA
.
reduce
(
+
,
dws
.
cs
;
dims
=
(
1
,
2
,
3
))),
lp
.
iL
[
end
])
return
nothing
return
reshape
(
Array
(
CUDA
.
reduce
(
+
,
dws
.
cs
;
dims
=
(
1
,
2
,
3
))),
lp
.
iL
[
end
])
end
end
function
kV
!
(
kv
,
dws
::
DiracWorkspace
,
prop
::
AbstractArray
,
lp
::
SpaceParm
)
function
kV
(
prop
::
AbstractArray
,
dws
::
DiracWorkspace
,
lp
::
SpaceParm
)
function
krnl_kV
!
(
sum
,
pro
)
function
krnl_kV
!
(
sum
,
pro
)
b
=
Int64
(
CUDA
.
threadIdx
()
.
x
)
b
=
Int64
(
CUDA
.
threadIdx
()
.
x
)
...
@@ -53,11 +51,11 @@ function kV!(kv, dws::DiracWorkspace, prop::AbstractArray,lp::SpaceParm)
...
@@ -53,11 +51,11 @@ function kV!(kv, dws::DiracWorkspace, prop::AbstractArray,lp::SpaceParm)
CUDA
.
@sync
begin
CUDA
.
@sync
begin
CUDA
.
@cuda
threads
=
lp
.
bsz
blocks
=
lp
.
rsz
krnl_kV!
(
dws
.
cs
,
prop
)
CUDA
.
@cuda
threads
=
lp
.
bsz
blocks
=
lp
.
rsz
krnl_kV!
(
dws
.
cs
,
prop
)
end
end
kv
+=
reshape
(
Array
(
CUDA
.
reduce
(
+
,
dws
.
cs
;
dims
=
(
1
,
2
,
3
))),
lp
.
iL
[
end
])
return
nothing
return
reshape
(
Array
(
CUDA
.
reduce
(
+
,
dws
.
cs
;
dims
=
(
1
,
2
,
3
))),
lp
.
iL
[
end
])
end
end
function
lV
!
(
lv
,
dws
::
DiracWorkspace
,
prop
::
AbstractArray
,
lp
::
SpaceParm
)
function
lV
(
prop
::
AbstractArray
,
dws
::
DiracWorkspace
,
lp
::
SpaceParm
)
function
krnl_lV
!
(
sum
,
pro
)
function
krnl_lV
!
(
sum
,
pro
)
b
=
Int64
(
CUDA
.
threadIdx
()
.
x
)
b
=
Int64
(
CUDA
.
threadIdx
()
.
x
)
...
@@ -73,6 +71,6 @@ function lV!(lv, dws::DiracWorkspace, prop::AbstractArray,lp::SpaceParm)
...
@@ -73,6 +71,6 @@ function lV!(lv, dws::DiracWorkspace, prop::AbstractArray,lp::SpaceParm)
CUDA
.
@sync
begin
CUDA
.
@sync
begin
CUDA
.
@cuda
threads
=
lp
.
bsz
blocks
=
lp
.
rsz
krnl_lV!
(
dws
.
cs
,
prop
)
CUDA
.
@cuda
threads
=
lp
.
bsz
blocks
=
lp
.
rsz
krnl_lV!
(
dws
.
cs
,
prop
)
end
end
lv
+=
reshape
(
Array
(
CUDA
.
reduce
(
+
,
dws
.
cs
;
dims
=
(
1
,
2
,
3
))),
lp
.
iL
[
end
])
return
nothing
return
reshape
(
Array
(
CUDA
.
reduce
(
+
,
dws
.
cs
;
dims
=
(
1
,
2
,
3
))),
lp
.
iL
[
end
])
end
end
This diff is collapsed.
Click to expand it.
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment