[TP comm overlap unit test]CUDA Error: misaligned address error when testing with recent cublas (or pytorch container)

[TP comm overlap unit test]CUDA Error: misaligned address error when testing with recent cublas (or pytorch container) #4406