Skip to content

Karsa Zoltán István / politopok

Go to a project
Commit 22e6404c by Zoltan Karsa
Options

fix block size problem

parent a6e0f9e9
Showing with 4 additions and 4 deletions
genax.py
...@@ -63,8 +63,8 @@ def angles_alap(anglestopick, plot = False): ...@@ -63,8 +63,8 @@ def angles_alap(anglestopick, plot = False):
m = anglestopick.size m = anglestopick.size
alpha_arr = cp.zeros((m, m), dtype=cp.float64) alpha_arr = cp.zeros((m, m), dtype=cp.float64)
beta_arr = cp.zeros((m, m), dtype=cp.float64) beta_arr = cp.zeros((m, m), dtype=cp.float64)
blocksize = int((m + 64 - 1) / 64)
parosit((m,), (m,), (anglestopick, anglestopick, alpha_arr, beta_arr, m, cp.pi)) parosit((blocksize,), (64,), (anglestopick, anglestopick, alpha_arr, beta_arr, m, cp.pi))
tompa_beta_arr = beta_arr[beta_arr > cp.pi] tompa_beta_arr = beta_arr[beta_arr > cp.pi]
tompa_beta_mpi_arr = tompa_beta_arr - cp.pi/2 tompa_beta_mpi_arr = tompa_beta_arr - cp.pi/2
...@@ -99,8 +99,8 @@ def angles_ratet(anglestopick, plot = False): ...@@ -99,8 +99,8 @@ def angles_ratet(anglestopick, plot = False):
m = anglestopick.size m = anglestopick.size
alpha_arr = cp.zeros((m, m), dtype=cp.float64) alpha_arr = cp.zeros((m, m), dtype=cp.float64)
beta_arr = cp.zeros((m, m), dtype=cp.float64) beta_arr = cp.zeros((m, m), dtype=cp.float64)
blocksize = int((m + 64 - 1) / 64)
parosit2((m,), (m,), (anglestopick, anglestopick, alpha_arr, beta_arr, m, cp.pi)) parosit2((blocksize,), (m,), (anglestopick, anglestopick, alpha_arr, beta_arr, m, cp.pi))
tompa_beta_arr = beta_arr[beta_arr > cp.pi] tompa_beta_arr = beta_arr[beta_arr > cp.pi]
tompa_beta_mpi_arr = tompa_beta_arr - cp.pi/2 tompa_beta_mpi_arr = tompa_beta_arr - cp.pi/2
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or sign in to comment