pytorch

11b0a84f - Enable LogSoftmax for SPMD tracing (#98380)

Enable LogSoftmax for SPMD tracing (#98380) Pull Request resolved: https://github.com/pytorch/pytorch/pull/98380 Approved by: https://github.com/wanchaol