Fix kernel args for small K (via Scott Gray)
Fix kernel args for small K (via Scott Gray)

NervanaSystems/neon,coufon/neon-distributed,dongjoon-hyun/neon,Jokeren/neon,NervanaSystems/neon,Jokeren/neon,dongjoon-hyun/neon,NervanaSystems/neon,coufon/neon-distributed,dongjoon-hyun/neon,coufon/neon-distributed,NervanaSystems/neon,Jokeren/neon,Jokeren/neon