太初平台causal_softmax算子

[xgqdut2016]

不知道为什么测试结束以后总是出现core dumped，此外发现如果把causal_softmax.py里面的数值精度对比改为assert torch.allclose(x.to("cpu"), ans.to("cpu"), atol=0, rtol=1e-2)，提前把数据转移到cpu上，这种做法可以通过测试，避开上面的问题，原因不详