Unify CosSinRoPE and ComplexRoPE YaRN computation (#3787) #10957

Job	Run time
set-matrix-rocm / set	4s
set-matrix / set	3s
build-test (cuda, mt-l-bx86iamx-176-1800-h100-8, cuda, 13.0, torchtitan-ubuntu-22.04-clang12, htt... / linux-job	10m 32s
build-test-rocm (rocm, linux.rocm.gpu.ecosystem.gfx950.8, rocm, 7.2, torchtitan-rocm-ubuntu-22.04... / linux-job	10m 59s
	21m 38s

Provide feedback