[Quantization] import torch.fx only if torch.__version__ >= 1.8

wanchen.swc

2023-03-30 0438b966d6d9ef89fdd7a8d112d6e03b34122b1e

1
2
3
4
5
6
7
8
9
10
import random
 
import numpy as np
import torch
 
 
def set_all_random_seed(seed: int):
    random.seed(seed)
    np.random.seed(seed)
    torch.random.manual_seed(seed)