Use explicit seed and only generate on cpu

samanklesaria · samanklesaria · commit 97ceaad9d558 · 2025-08-11T20:36:35.000Z
diff --git a/test/torchaudio_unittest/functional/kaldi_compatibility_test_impl.py b/test/torchaudio_unittest/functional/kaldi_compatibility_test_impl.py
@@ -18,7 +18,8 @@ def test_sliding_window_cmn(self):
             "norm_vars": False,
         }
 
-        tensor = torch.randn(40, 10, dtype=self.dtype, device=self.device)
+        torch.manual_seed(0)
+        tensor = torch.randn(40, 10, dtype=self.dtype).to(self.device)
         result = F.sliding_window_cmn(tensor, **kwargs)
         command = ["apply-cmvn-sliding"] + convert_args(**kwargs) + ["ark:-", "ark:-"]
         kaldi_result = run_kaldi(self.request, command, "ark", tensor)

Original file line number	Diff line number	Diff line change
`@@ -18,7 +18,8 @@ def test_sliding_window_cmn(self):`
`18`	`18`	`"norm_vars": False,`
`19`	`19`	`}`
`20`	`20`
`21`		`- tensor = torch.randn(40, 10, dtype=self.dtype, device=self.device)`
	`21`	`+ torch.manual_seed(0)`
	`22`	`+ tensor = torch.randn(40, 10, dtype=self.dtype).to(self.device)`
`22`	`23`	`result = F.sliding_window_cmn(tensor, **kwargs)`
`23`	`24`	`command = ["apply-cmvn-sliding"] + convert_args(**kwargs) + ["ark:-", "ark:-"]`
`24`	`25`	`kaldi_result = run_kaldi(self.request, command, "ark", tensor)`