laithsakka 2024-08-14 11:37:18 -07:00 committed by PyTorch MergeBot
parent dff388491b
commit 7673ee5456
17 changed files with 26 additions and 13 deletions

View File

@ -399,7 +399,7 @@ pr_time_benchmarks() {
TEST_REPORTS_DIR=$(pwd)/test/test-reports
mkdir -p "$TEST_REPORTS_DIR"
source benchmarks/dynamo/pr_time_benchmarks/benchmark_runner.sh "$TEST_REPORTS_DIR/pr_time_benchmarks_after.txt" "benchmarks/dynamo/pr_time_benchmarks/benchmarks"
PYTHONPATH=$(pwd)/benchmarks/dynamo/pr_time_benchmarks source benchmarks/dynamo/pr_time_benchmarks/benchmark_runner.sh "$TEST_REPORTS_DIR/pr_time_benchmarks_after.txt" "benchmarks/dynamo/pr_time_benchmarks/benchmarks"
echo "benchmark results on current PR: "
cat "$TEST_REPORTS_DIR/pr_time_benchmarks_after.txt"

View File

View File

@ -21,5 +21,5 @@ python_programs_dir=$2
for file in $python_programs_dir/*.py
do
# Execute the Python program and append the output to the output file
sudo env PATH="$PATH" python $file $output_file
python $file $output_file
done

View File

@ -1,7 +1,7 @@
import random
import sys
from benchmarks.dynamo.pr_time_benchmarks.benchmark_base import BenchmarkBase
from benchmark_base import BenchmarkBase
import torch

View File

@ -1,4 +1,3 @@
import operator_benchmark as op_bench
from pt import ( # noqa: F401
add_test,
ao_sparsifier_test,
@ -30,6 +29,8 @@ from pt import ( # noqa: F401
tensor_to_test,
)
import operator_benchmark as op_bench
if __name__ == "__main__":
op_bench.benchmark_runner.main()

View File

@ -1,4 +1,3 @@
import operator_benchmark as op_bench
from pt import ( # noqa: F401
qactivation_test,
qarithmetic_test,
@ -22,6 +21,8 @@ from pt import ( # noqa: F401
qunary_test,
)
import operator_benchmark as op_bench
if __name__ == "__main__":
op_bench.benchmark_runner.main()

View File

@ -1,8 +1,9 @@
import benchmark_all_other_test # noqa: F401
import benchmark_all_quantized_test # noqa: F401
import operator_benchmark as op_bench
from pt import unary_test # noqa: F401
import operator_benchmark as op_bench
if __name__ == "__main__":
op_bench.benchmark_runner.main()

View File

@ -1,6 +1,7 @@
import operator_benchmark as op_bench
from pt import configs
import operator_benchmark as op_bench
import torch
import torch.nn as nn

View File

@ -1,7 +1,8 @@
import numpy
import operator_benchmark as op_bench
from pt import configs
import operator_benchmark as op_bench
import torch

View File

@ -1,4 +1,5 @@
import numpy
import operator_benchmark as op_bench
import torch

View File

@ -1,4 +1,5 @@
import numpy
import operator_benchmark as op_bench
import torch

View File

@ -1,6 +1,7 @@
import operator_benchmark as op_bench
from pt import configs
import operator_benchmark as op_bench
import torch
import torch.nn as nn

View File

@ -1,7 +1,8 @@
import numpy
import operator_benchmark as op_bench
from pt import configs
import operator_benchmark as op_bench
import torch
import torch.ao.nn.qat as nnqat
from torch.ao.quantization import default_embedding_qat_qconfig

View File

@ -1,6 +1,7 @@
import operator_benchmark as op_bench
from pt import configs
import operator_benchmark as op_bench
import torch
import torch.ao.nn.quantized as nnq

View File

@ -1,6 +1,7 @@
from typing import Optional
import numpy as np
import operator_benchmark as op_bench
import torch

View File

@ -1,7 +1,8 @@
import numpy
import operator_benchmark as op_bench
from pt import configs
import operator_benchmark as op_bench
import torch
import torch.ao.nn.quantized as nnq

View File

@ -1,6 +1,7 @@
import operator_benchmark as op_bench
from pt import configs
import operator_benchmark as op_bench
import torch
import torch.ao.nn.quantized as nnq
import torch.ao.nn.quantized.dynamic as nnqd