Skip to content
Closed
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 2 additions & 0 deletions tritonparse/__main__.py
Original file line number Diff line number Diff line change
@@ -1,3 +1,5 @@
# Copyright (c) Meta Platforms, Inc. and affiliates.

from .cli import main


Expand Down
2 changes: 2 additions & 0 deletions tritonparse/context_manager.py
Original file line number Diff line number Diff line change
@@ -1,3 +1,5 @@
# Copyright (c) Meta Platforms, Inc. and affiliates.

import os
import shutil
import tempfile
Expand Down
2 changes: 2 additions & 0 deletions tritonparse/event_diff.py
Original file line number Diff line number Diff line change
@@ -1,3 +1,5 @@
# Copyright (c) Meta Platforms, Inc. and affiliates.

import json
from collections import defaultdict
from typing import Any, Dict, List, Tuple
Expand Down
2 changes: 2 additions & 0 deletions tritonparse/ir_analysis.py
Original file line number Diff line number Diff line change
@@ -1,3 +1,5 @@
# Copyright (c) Meta Platforms, Inc. and affiliates.

import logging

from .sourcemap_utils import load_ir_contents
Expand Down
2 changes: 2 additions & 0 deletions tritonparse/ir_parser.py
Original file line number Diff line number Diff line change
@@ -1,3 +1,5 @@
# Copyright (c) Meta Platforms, Inc. and affiliates.

import logging
import os
import re
Expand Down
2 changes: 2 additions & 0 deletions tritonparse/mapper.py
Original file line number Diff line number Diff line change
@@ -1,3 +1,5 @@
# Copyright (c) Meta Platforms, Inc. and affiliates.

import logging
from collections import defaultdict
from typing import Any, Dict, List, Tuple
Expand Down
2 changes: 2 additions & 0 deletions tritonparse/reproducer/cli.py
Original file line number Diff line number Diff line change
@@ -1,3 +1,5 @@
# Copyright (c) Meta Platforms, Inc. and affiliates.

import argparse

from tritonparse.reproducer.types import KernelImportMode
Expand Down
2 changes: 2 additions & 0 deletions tritonparse/reproducer/function_extractor.py
Original file line number Diff line number Diff line change
@@ -1,3 +1,5 @@
# Copyright (c) Meta Platforms, Inc. and affiliates.

"""
Function extractor for reproducer utility functions.

Expand Down
2 changes: 2 additions & 0 deletions tritonparse/reproducer/ingestion/ndjson.py
Original file line number Diff line number Diff line change
@@ -1,3 +1,5 @@
# Copyright (c) Meta Platforms, Inc. and affiliates.

from dataclasses import dataclass
from typing import Any, Dict, List, Optional, Tuple

Expand Down
2 changes: 2 additions & 0 deletions tritonparse/reproducer/orchestrator.py
Original file line number Diff line number Diff line change
@@ -1,3 +1,5 @@
# Copyright (c) Meta Platforms, Inc. and affiliates.

from pathlib import Path
from typing import Optional

Expand Down
2 changes: 2 additions & 0 deletions tritonparse/reproducer/placeholder_replacer.py
Original file line number Diff line number Diff line change
@@ -1,3 +1,5 @@
# Copyright (c) Meta Platforms, Inc. and affiliates.

from abc import ABC

from typing import Any, Dict, Protocol
Expand Down
2 changes: 2 additions & 0 deletions tritonparse/reproducer/templates/loader.py
Original file line number Diff line number Diff line change
@@ -1,3 +1,5 @@
# Copyright (c) Meta Platforms, Inc. and affiliates.

from importlib.resources import files as pkg_files
from pathlib import Path
from typing import List
Expand Down
2 changes: 2 additions & 0 deletions tritonparse/reproducer/types.py
Original file line number Diff line number Diff line change
@@ -1,3 +1,5 @@
# Copyright (c) Meta Platforms, Inc. and affiliates.

from enum import Enum


Expand Down
2 changes: 2 additions & 0 deletions tritonparse/reproducer/utils.py
Original file line number Diff line number Diff line change
@@ -1,3 +1,5 @@
# Copyright (c) Meta Platforms, Inc. and affiliates.

import importlib
import importlib.util
import json
Expand Down
1 change: 1 addition & 0 deletions tritonparse/shared_vars.py
Original file line number Diff line number Diff line change
@@ -1,3 +1,4 @@
# Copyright (c) Meta Platforms, Inc. and affiliates.
# We'd like to sperate structured logging module and tritonparse module as much as possible. So, put the shared variables here.
import os

Expand Down
2 changes: 2 additions & 0 deletions tritonparse/sourcemap_utils.py
Original file line number Diff line number Diff line change
@@ -1,3 +1,5 @@
# Copyright (c) Meta Platforms, Inc. and affiliates.

import logging
from typing import Any, Dict, List

Expand Down
2 changes: 2 additions & 0 deletions tritonparse/tools/decompress_bin_ndjson.py
Original file line number Diff line number Diff line change
@@ -1,4 +1,6 @@
#!/usr/bin/env python3
# Copyright (c) Meta Platforms, Inc. and affiliates.

"""
Script to decompress .bin.ndjson files back to regular .ndjson format.

Expand Down
2 changes: 2 additions & 0 deletions tritonparse/tools/format_fix.py
Original file line number Diff line number Diff line change
@@ -1,4 +1,6 @@
#!/usr/bin/env python3
# Copyright (c) Meta Platforms, Inc. and affiliates.

"""
Format fix script for tritonparse project.

Expand Down
2 changes: 2 additions & 0 deletions tritonparse/tools/load_tensor.py
Original file line number Diff line number Diff line change
@@ -1,4 +1,6 @@
#!/usr/bin/env python3
# Copyright (c) Meta Platforms, Inc. and affiliates.

"""
Simple tensor loading utility for tritonparse saved tensors.
Usage:
Expand Down
2 changes: 2 additions & 0 deletions tritonparse/tools/prettify_ndjson.py
Original file line number Diff line number Diff line change
@@ -1,4 +1,6 @@
#!/usr/bin/env python3
# Copyright (c) Meta Platforms, Inc. and affiliates.

"""
Convert an NDJSON file to a prettified JSON file.

Expand Down
2 changes: 2 additions & 0 deletions tritonparse/trace_processor.py
Original file line number Diff line number Diff line change
@@ -1,3 +1,5 @@
# Copyright (c) Meta Platforms, Inc. and affiliates.

import gzip
import json
import logging
Expand Down