-
Notifications
You must be signed in to change notification settings - Fork 13
Expand file tree
/
Copy pathtest_examples.py
More file actions
67 lines (41 loc) · 1.87 KB
/
test_examples.py
File metadata and controls
67 lines (41 loc) · 1.87 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
# SPDX-FileCopyrightText: Copyright (c) 2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
# SPDX-License-Identifier: Apache-2.0
import importlib
import pytest
def test_00_minimal() -> None:
minimal = importlib.import_module("examples.00_minimal")
minimal.main()
def test_01_compare_throughput() -> None:
compare_throughput = importlib.import_module("examples.01_compare_throughput")
compare_throughput.main()
def test_02_parameter_sweep() -> None:
parameter_sweep = importlib.import_module("examples.02_parameter_sweep")
parameter_sweep.main()
def test_03_custom_metrics() -> None:
custom_metrics = importlib.import_module("examples.03_custom_metrics")
custom_metrics.main()
def test_04_multi_parameter() -> None:
multi_parameter = importlib.import_module("examples.04_multi_parameter")
multi_parameter.main()
def test_05_subplots() -> None:
subplots = importlib.import_module("examples.05_subplots")
subplots.main()
def test_06_plot_customization() -> None:
plot_customization = importlib.import_module("examples.06_plot_customization")
plot_customization.main()
def test_07_triton_minimal() -> None:
pytest.importorskip("triton")
triton_minimal = importlib.import_module("examples.07_triton_minimal")
triton_minimal.main()
def test_08_multiple_metrics() -> None:
multiple_metrics = importlib.import_module("examples.08_multiple_metrics")
multiple_metrics.main()
def test_09_advanced_metric_custom() -> None:
advanced_custom = importlib.import_module("examples.09_advanced_metric_custom")
advanced_custom.main()
def test_10_combine_kernel_metrics() -> None:
combine_metrics = importlib.import_module("examples.10_combine_kernel_metrics")
combine_metrics.main()
def test_11_output_csv() -> None:
output_csv = importlib.import_module("examples.11_output_csv")
output_csv.main()