Upload bin/proton with huggingface_hub
Browse files- bin/proton +7 -0
bin/proton
ADDED
|
@@ -0,0 +1,7 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
#!/home/crhon/pytorch_rocm_env/bin/python3
|
| 2 |
+
import sys
|
| 3 |
+
from triton.profiler.proton import main
|
| 4 |
+
if __name__ == '__main__':
|
| 5 |
+
if sys.argv[0].endswith('.exe'):
|
| 6 |
+
sys.argv[0] = sys.argv[0][:-4]
|
| 7 |
+
sys.exit(main())
|