Skip to content

Commit 1a0b350

Browse files
FindHaofacebook-github-bot
authored andcommitted
Setup internal ci tests
Summary: **Add main block to test_tritonparse.py to enable Buck execution** Added a `if __name__ == "__main__":` block at the end of `test_tritonparse.py` that calls `pytest.main([__file__, "-v"])`. This allows the test file to be executed directly by Buck's `python_unittest_remote_gpu` target while preserving the existing pytest format and functionality. **Changes:** * Added main execution block to [`**test_tritonparse.py**`] * No changes to existing test logic or BUCK configuration * Maintains compatibility with both Buck execution and direct pytest execution Reviewed By: xuzhao9 Differential Revision: D77940958 fbshipit-source-id: 403e1291ec5976c2f218de59e266bad934378059
1 parent 44bf2f1 commit 1a0b350

File tree

1 file changed

+8
-0
lines changed

1 file changed

+8
-0
lines changed

tests/test_tritonparse.py

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -15,12 +15,20 @@
1515
from typing import Any, Union
1616

1717
import torch
18+
19+
# @manual=//triton:triton
1820
import triton
21+
22+
# @manual=//triton:triton
1923
import triton.language as tl
2024

2125
import tritonparse.structured_logging
2226
import tritonparse.utils
27+
28+
# @manual=//triton:triton
2329
from triton.compiler import ASTSource, IRSource
30+
31+
# @manual=//triton:triton
2432
from triton.knobs import CompileTimes
2533
from tritonparse.structured_logging import convert, extract_python_source_info
2634

0 commit comments

Comments
 (0)