Skip to content

Commit ff50319

Browse files
kavin-sai-krishnaDhanshrePriya
authored andcommitted
Changed default periodic value to True
1 parent 31747eb commit ff50319

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

python/tvm/relax/frontend/torch/exported_program_translator.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -26,6 +26,8 @@
2626
import tvm
2727
from tvm import relax
2828

29+
import tvm.relax
30+
tvm.relax.get_default_pipeline
2931
from .base_fx_graph_translator import BaseFXGraphImporter
3032

3133

@@ -288,7 +290,7 @@ def _hamming_window(self, node: fx.Node) -> relax.Var:
288290
args = self.retrieve_args(node)
289291

290292
window_size = args[0]
291-
periodic = args[1] if len(args) > 1 else False
293+
periodic = args[1] if len(args) > 1 else True
292294
alpha = args[2] if len(args) > 2 else 0.54
293295
beta = args[3] if len(args) > 3 else 0.46
294296
dtype = node.kwargs.get("dtype", "float")
@@ -545,6 +547,7 @@ def create_convert_map(
545547
"hamming_window.periodic": self._hamming_window,
546548
"hamming_window.periodic_alpha": self._hamming_window,
547549
"hamming_window.periodic_alpha_beta": self._hamming_window,
550+
"hamming_window.default": self._hamming_window,
548551
"index_select.default": self._index_select,
549552
"lift_fresh_copy.default": self._to_copy,
550553
"linspace.default": self._linspace,

0 commit comments

Comments
 (0)