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
|
/// Basic uniform random distribution
|
||||||
InitialDistribution(int64_t _seed = 700) : seed(_seed) {
|
InitialDistribution(int64_t _seed = 700) : seed(_seed) {
|
||||||
dist_A.set_uniform(-8, 8);
|
dist_A.set_uniform(-4, 4);
|
||||||
dist_B.set_uniform(-8, 8);
|
dist_B.set_uniform(-4, 4);
|
||||||
dist_C.set_uniform(-8, 8);
|
dist_C.set_uniform(-4, 4);
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Extracts initial distribution from command line arguments
|
/// Extracts initial distribution from command line arguments
|
||||||
InitialDistribution(cutlass::CommandLine const &args) {
|
InitialDistribution(cutlass::CommandLine const &args) {
|
||||||
// Set initial values
|
// Set initial values
|
||||||
seed = 700;
|
seed = 700;
|
||||||
dist_A.set_uniform(-8, 8);
|
dist_A.set_uniform(-4, 4);
|
||||||
dist_B.set_uniform(-8, 8);
|
dist_B.set_uniform(-4, 4);
|
||||||
dist_C.set_uniform(-8, 8);
|
dist_C.set_uniform(-4, 4);
|
||||||
|
|
||||||
// Update with command line arguments
|
// Update with command line arguments
|
||||||
args.get_cmd_line_argument("seed", seed, seed);
|
args.get_cmd_line_argument("seed", seed, seed);
|
||||||
|
Loading…
Reference in New Issue
Block a user