Skip to content

Commit 22725d3

Browse files
authored
[BENCHMARKS] Use setuptools.find_packages to include all python submodules (#5177)
It's needed for example for `configs` submodule. Signed-off-by: Anatoly Myachev <[email protected]>
1 parent c0af02d commit 22725d3

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

benchmarks/setup.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -7,7 +7,7 @@
77
# https://github.com/pypa/setuptools/discussions/2838
88
from distutils import log # pylint: disable=[deprecated-module]
99

10-
from setuptools import setup, Extension
10+
from setuptools import setup, Extension, find_packages
1111
from setuptools.command.build_ext import build_ext as _build_ext
1212

1313
import torch
@@ -112,7 +112,7 @@ def get_git_commit_hash(length=8):
112112
setup(
113113
name="triton-kernels-benchmark",
114114
version="3.3.0" + get_git_commit_hash(),
115-
packages=["triton_kernels_benchmark"],
115+
packages=find_packages(),
116116
install_requires=[
117117
"torch>=2.6",
118118
"pandas",

benchmarks/triton_kernels_benchmark/benchmark_utils.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -19,7 +19,7 @@
1919
MarkArgs,
2020
)
2121
from triton_kernels_benchmark.benchmark_shapes_parser import ShapePatternParser
22-
from triton_kernels_benchmark.benchmark_config_templates import CONFIGS
22+
from triton_kernels_benchmark.configs.benchmark_config_templates import CONFIGS
2323

2424

2525
@dataclass

0 commit comments

Comments
 (0)