polygamma is broken on windows cuda

pytorch / functorch

functorch is JAX-like composable function transforms for PyTorch.

https://pytorch.org/functorch/

BSD 3-Clause "New" or "Revised" License

1.38k stars 102 forks source link

polygamma is broken on windows cuda #934

Closed samdow closed 1 year ago

samdow commented 1 year ago

Job: https://app.circleci.com/pipelines/github/pytorch/functorch/3375/workflows/fc5892a5-ca3a-44fd-912a-612123b004c2/jobs/26445/tests

All calls are failing on the call with "default_program(304): error: identifier "zeta" is undefined" and a dump with a lot of code in it. It looks like included in the dump is polygamma_vectorized4_kernel. Not certain what's happening or how polygamma is passing tests upstream since it doesn't look like this function is in functorch

samdow commented 1 year ago

Seems to have been a transient error