cp kernels_req_3/layernorm.cuh kernels/layernorm.cuh
cp kernels_req_3/softmax.cuh kernels/softmax.cuh
cp kernels_req_3/attention.cuh kernels/attention.cuh 

make test_gpt2

git checkout kernels/softmax.cuh
git checkout kernels/layernorm.cuh
git checkout kernels/attention.cuh