Cool! I am blind, I guess.
@jick, -keep option should leave ptx files, though I don’t know where they’ll end up during pytorch build process. Alternatively, I think you can use cuobjdump to dump ptx from compiled libAten.