Skip to content

Commit 0b2ebcd

Browse files
committed
Added launch bounds to fix launch resource error for Lion.
1 parent ed6f3eb commit 0b2ebcd

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

csrc/kernels.cu

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1260,6 +1260,7 @@ kPreconditionOptimizerStatic8bit1State(T* p, T* __restrict__ const g, unsigned c
12601260

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

0 commit comments

Comments
 (0)