Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
OpenDAS
nni
Commits
f908be2c
"src/include/blockwise_gemm.hpp" did not exist on "00899f191b1ffbe24bdaf982d8ebf6a4e29697c3"
Unverified
Commit
f908be2c
authored
Dec 03, 2021
by
Zhenhua Han
Committed by
GitHub
Dec 03, 2021
Browse files
[retiarii] use fixed seeds for unit test of CGO engine (#4361)
parent
1abaf627
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
3 additions
and
0 deletions
+3
-0
test/ut/retiarii/test_cgo_engine.py
test/ut/retiarii/test_cgo_engine.py
+3
-0
No files found.
test/ut/retiarii/test_cgo_engine.py
View file @
f908be2c
...
...
@@ -5,6 +5,7 @@ import unittest
import
time
import
torch
import
torch.nn
as
nn
from
pytorch_lightning.utilities.seed
import
seed_everything
from
pathlib
import
Path
...
...
@@ -137,6 +138,8 @@ def _reset():
nni
.
runtime
.
platform
.
test
.
_last_metric
=
None
nni
.
retiarii
.
integration_api
.
_advisor
=
None
nni
.
retiarii
.
execution
.
api
.
_execution_engine
=
None
seed_everything
(
42
)
def
_new_trainer
():
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment