Added launch bounds to fix launch resource error for Lion.

2023-04-11 08:37:02 -07:00 · 2023-04-11 08:37:02 -07:00 · 0b2ebcdab9
commit 0b2ebcdab9
parent ed6f3eb146
1 changed files with 1 additions and 0 deletions
--- a/csrc/kernels.cu
+++ b/csrc/kernels.cu
@ -1260,6 +1260,7 @@ kPreconditionOptimizerStatic8bit1State(T* p, T* __restrict__ const g, unsigned c

 template<typename T, int OPTIMIZER>
 __global__ void
+__launch_bounds__(1024, 1)
 kOptimizerStatic8bit1State(T* p, T* const g, unsigned char* state1,
                const float *unorm, const float max_unorm, const float param_norm,
                const float beta1, const float beta2,