288 lines
5.5 KiB
JSON
288 lines
5.5 KiB
JSON
{
|
|
"passed": true,
|
|
"reference_commit": "fd3cf0140de03f7619ac95b5cfed05b711ff9e22",
|
|
"source_sha256": "4463b8a108f678c166755101d95f188179f54143305953d78147de0389b80ebc",
|
|
"torch": "2.15.0+cpu",
|
|
"numpy": "2.5.4",
|
|
"implementation_sha256": "48a2a97d0445dc6e08ac7f27a935637a7553417d0982eb94122d6cb02bbb499c",
|
|
"tests": [
|
|
{
|
|
"test": "layer0_rmsnorm",
|
|
"shape": [
|
|
2,
|
|
7,
|
|
16
|
|
],
|
|
"max_abs_error": 2.384185791015625e-07,
|
|
"atol": 2e-05,
|
|
"rtol": 2e-05,
|
|
"passed": true
|
|
},
|
|
{
|
|
"test": "layer0_mixer",
|
|
"shape": [
|
|
2,
|
|
7,
|
|
16
|
|
],
|
|
"max_abs_error": 1.862645149230957e-07,
|
|
"atol": 2e-05,
|
|
"rtol": 2e-05,
|
|
"passed": true
|
|
},
|
|
{
|
|
"test": "layer0_mlp",
|
|
"shape": [
|
|
2,
|
|
7,
|
|
16
|
|
],
|
|
"max_abs_error": 2.9802322387695312e-08,
|
|
"atol": 2e-05,
|
|
"rtol": 2e-05,
|
|
"passed": true
|
|
},
|
|
{
|
|
"test": "layer0_complete",
|
|
"shape": [
|
|
2,
|
|
7,
|
|
16
|
|
],
|
|
"max_abs_error": 2.384185791015625e-07,
|
|
"atol": 2e-05,
|
|
"rtol": 2e-05,
|
|
"passed": true
|
|
},
|
|
{
|
|
"test": "layer1_rmsnorm",
|
|
"shape": [
|
|
2,
|
|
7,
|
|
16
|
|
],
|
|
"max_abs_error": 2.384185791015625e-07,
|
|
"atol": 2e-05,
|
|
"rtol": 2e-05,
|
|
"passed": true
|
|
},
|
|
{
|
|
"test": "attention_probabilities",
|
|
"shape": [
|
|
2,
|
|
2,
|
|
7,
|
|
7
|
|
],
|
|
"max_abs_error": 1.1920928955078125e-07,
|
|
"atol": 2e-05,
|
|
"rtol": 2e-05,
|
|
"passed": true
|
|
},
|
|
{
|
|
"test": "layer1_mixer",
|
|
"shape": [
|
|
2,
|
|
7,
|
|
16
|
|
],
|
|
"max_abs_error": 5.21540641784668e-08,
|
|
"atol": 2e-05,
|
|
"rtol": 2e-05,
|
|
"passed": true
|
|
},
|
|
{
|
|
"test": "layer1_mlp",
|
|
"shape": [
|
|
2,
|
|
7,
|
|
16
|
|
],
|
|
"max_abs_error": 4.470348358154297e-08,
|
|
"atol": 2e-05,
|
|
"rtol": 2e-05,
|
|
"passed": true
|
|
},
|
|
{
|
|
"test": "layer1_complete",
|
|
"shape": [
|
|
2,
|
|
7,
|
|
16
|
|
],
|
|
"max_abs_error": 2.384185791015625e-07,
|
|
"atol": 2e-05,
|
|
"rtol": 2e-05,
|
|
"passed": true
|
|
},
|
|
{
|
|
"test": "torch_recurrent_gated_delta_rule_length1_output",
|
|
"shape": [
|
|
2,
|
|
1,
|
|
2,
|
|
5
|
|
],
|
|
"max_abs_error": 4.470348358154297e-08,
|
|
"atol": 2e-05,
|
|
"rtol": 2e-05,
|
|
"passed": true
|
|
},
|
|
{
|
|
"test": "torch_recurrent_gated_delta_rule_length1_state",
|
|
"shape": [
|
|
2,
|
|
2,
|
|
4,
|
|
5
|
|
],
|
|
"max_abs_error": 5.960464477539063e-08,
|
|
"atol": 2e-05,
|
|
"rtol": 2e-05,
|
|
"passed": true
|
|
},
|
|
{
|
|
"test": "torch_chunk_gated_delta_rule_length1_output",
|
|
"shape": [
|
|
2,
|
|
1,
|
|
2,
|
|
5
|
|
],
|
|
"max_abs_error": 2.9802322387695312e-08,
|
|
"atol": 2e-05,
|
|
"rtol": 2e-05,
|
|
"passed": true
|
|
},
|
|
{
|
|
"test": "torch_chunk_gated_delta_rule_length1_state",
|
|
"shape": [
|
|
2,
|
|
2,
|
|
4,
|
|
5
|
|
],
|
|
"max_abs_error": 5.960464477539063e-08,
|
|
"atol": 2e-05,
|
|
"rtol": 2e-05,
|
|
"passed": true
|
|
},
|
|
{
|
|
"test": "torch_recurrent_gated_delta_rule_length7_output",
|
|
"shape": [
|
|
2,
|
|
7,
|
|
2,
|
|
5
|
|
],
|
|
"max_abs_error": 5.960464477539063e-08,
|
|
"atol": 2e-05,
|
|
"rtol": 2e-05,
|
|
"passed": true
|
|
},
|
|
{
|
|
"test": "torch_recurrent_gated_delta_rule_length7_state",
|
|
"shape": [
|
|
2,
|
|
2,
|
|
4,
|
|
5
|
|
],
|
|
"max_abs_error": 5.960464477539063e-08,
|
|
"atol": 2e-05,
|
|
"rtol": 2e-05,
|
|
"passed": true
|
|
},
|
|
{
|
|
"test": "torch_chunk_gated_delta_rule_length7_output",
|
|
"shape": [
|
|
2,
|
|
7,
|
|
2,
|
|
5
|
|
],
|
|
"max_abs_error": 7.450580596923828e-08,
|
|
"atol": 2e-05,
|
|
"rtol": 2e-05,
|
|
"passed": true
|
|
},
|
|
{
|
|
"test": "torch_chunk_gated_delta_rule_length7_state",
|
|
"shape": [
|
|
2,
|
|
2,
|
|
4,
|
|
5
|
|
],
|
|
"max_abs_error": 1.7881393432617188e-07,
|
|
"atol": 2e-05,
|
|
"rtol": 2e-05,
|
|
"passed": true
|
|
},
|
|
{
|
|
"test": "torch_recurrent_gated_delta_rule_length33_output",
|
|
"shape": [
|
|
2,
|
|
33,
|
|
2,
|
|
5
|
|
],
|
|
"max_abs_error": 1.1920928955078125e-07,
|
|
"atol": 2e-05,
|
|
"rtol": 2e-05,
|
|
"passed": true
|
|
},
|
|
{
|
|
"test": "torch_recurrent_gated_delta_rule_length33_state",
|
|
"shape": [
|
|
2,
|
|
2,
|
|
4,
|
|
5
|
|
],
|
|
"max_abs_error": 2.9802322387695312e-08,
|
|
"atol": 2e-05,
|
|
"rtol": 2e-05,
|
|
"passed": true
|
|
},
|
|
{
|
|
"test": "torch_chunk_gated_delta_rule_length33_output",
|
|
"shape": [
|
|
2,
|
|
33,
|
|
2,
|
|
5
|
|
],
|
|
"max_abs_error": 5.364418029785156e-07,
|
|
"atol": 2e-05,
|
|
"rtol": 2e-05,
|
|
"passed": true
|
|
},
|
|
{
|
|
"test": "torch_chunk_gated_delta_rule_length33_state",
|
|
"shape": [
|
|
2,
|
|
2,
|
|
4,
|
|
5
|
|
],
|
|
"max_abs_error": 3.5762786865234375e-07,
|
|
"atol": 2e-05,
|
|
"rtol": 2e-05,
|
|
"passed": true
|
|
},
|
|
{
|
|
"test": "complete_two_layer_text_forward",
|
|
"shape": [
|
|
2,
|
|
7,
|
|
16
|
|
],
|
|
"max_abs_error": 3.5762786865234375e-07,
|
|
"atol": 2e-05,
|
|
"rtol": 2e-05,
|
|
"passed": true
|
|
}
|
|
],
|
|
"limitations": "FP32 CPU semantics on tiny synthetic configurations; not full-checkpoint BF16 kernel equivalence"
|
|
} |