OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-06 12:20:52 +01:00
Code Issues Actions 38 Packages Projects Releases Wiki Activity

Fix Undefined variable in QInterpolateBenchmark

Browse Source 
Pull Request resolved: https://github.com/pytorch/pytorch/pull/73130
Approved by: https://github.com/malfet
This commit is contained in:
Sergii Dymchenko 2022-03-09 00:14:15 +00:00 committed by PyTorch MergeBot
parent dc38326f1d
commit 5b011fc6eb
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
benchmarks/operator_benchmark/pt/qinterpolate_test.py
View File

@ -44,7 +44,7 @@ class QInterpolateBenchmark(op_bench.TorchBenchmarkBase):
zero_point=zero_point,
dtype=dtype)
if not contig:
permute_dims = list(range(q_input.ndim))[::-1]
permute_dims = list(range(self.q_input.ndim))[::-1]
self.q_input = self.q_input.permute(permute_dims)
self.inputs = {