pytorch-nightly-checker.py

import pathlib
import importlib.util

s=importlib.util.find_spec('triton')
affected=any(x.name == 'triton' for x in (pathlib.Path(s.submodule_search_locations[0] if s is not None else '/' ) / 'runtime').glob('*'))
print('You are {}affected'.format('' if affected else 'not '))