torch-mlir/python/npcomp/smoketest.py