Fix torch, new kernel