"src/include/blockwise_gemm.hpp" did not exist on "00899f191b1ffbe24bdaf982d8ebf6a4e29697c3"
Unverified Commit f908be2c authored by Zhenhua Han's avatar Zhenhua Han Committed by GitHub
Browse files

[retiarii] use fixed seeds for unit test of CGO engine (#4361)

parent 1abaf627
......@@ -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():
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment