Update custom_ops.py#315

matt-graham merged 3 commits intomainastro-informatics/s2fft:mainfrom

km/s2ai-compatabilityastro-informatics/s2fft:km/s2ai-compatabilityCopy head branch name to clipboard