I have seen in several places examples using PYTORCH_FUSION_DEBUG=1 to retrieve the source of the fused kernels (for example here [JIT] Fusion of Dropout without constant is_training parameter is unsuccessful · Issue #24032 · pytorch/pytorch · GitHub 4), I am assuming this dumps it to stdout, but when run with this variable set I see nothing.
Could anyone give advice on how to get this working? Perhaps I need to compile PyTorch with specific flags set?