Reduced range of random values to avoid bit-level inconsistencies for large matrices.
This commit is contained in:
parent
77d1e0ca81
commit
0826572c4c
@ -263,18 +263,18 @@ struct InitialDistribution {
|
||||
|
||||
/// Basic uniform random distribution
|
||||
InitialDistribution(int64_t _seed = 700) : seed(_seed) {
|
||||
dist_A.set_uniform(-8, 8);
|
||||
dist_B.set_uniform(-8, 8);
|
||||
dist_C.set_uniform(-8, 8);
|
||||
dist_A.set_uniform(-4, 4);
|
||||
dist_B.set_uniform(-4, 4);
|
||||
dist_C.set_uniform(-4, 4);
|
||||
}
|
||||
|
||||
/// Extracts initial distribution from command line arguments
|
||||
InitialDistribution(cutlass::CommandLine const &args) {
|
||||
// Set initial values
|
||||
seed = 700;
|
||||
dist_A.set_uniform(-8, 8);
|
||||
dist_B.set_uniform(-8, 8);
|
||||
dist_C.set_uniform(-8, 8);
|
||||
dist_A.set_uniform(-4, 4);
|
||||
dist_B.set_uniform(-4, 4);
|
||||
dist_C.set_uniform(-4, 4);
|
||||
|
||||
// Update with command line arguments
|
||||
args.get_cmd_line_argument("seed", seed, seed);
|
||||
|
Loading…
Reference in New Issue
Block a user