[Flux.1] improve pos embed for ascend npu by computing on npu

[Flux.1] improve pos embed for ascend npu by computing on npu #35561