update kernel parameters for nvidia

seeder-research · May 9, 2022 · 38e62c0 · 38e62c0
1 parent 10f8892
commit 38e62c0
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/opencl/init.go b/opencl/init.go
@@ -226,6 +226,7 @@ func Init(gpu int) {
 	ClMaxWGNum = ClCUnits
 	ClTotalPE = ClWGSize[2] * ClCUnits
 	if GPUVend == 1 { // Nvidia
+		ClTotalPE *= 2
 		if ClMaxWGSize > ClTotalPE {
 			ClMaxWGNum = 1
 			ClMaxWGSize = ClTotalPE

diff --git a/opencl64/init.go b/opencl64/init.go
@@ -229,6 +229,7 @@ func Init(gpu int) {
 	ClMaxWGNum = ClCUnits
 	ClTotalPE = ClWGSize[2] * ClCUnits
 	if GPUVend == 1 { // Nvidia
+		ClTotalPE *= 2
 		if ClMaxWGSize > ClTotalPE {
 			ClMaxWGNum = 1
 			ClMaxWGSize = ClTotalPE