Compare commits

...

2 Commits

Author SHA1 Message Date
github-actions[bot]
ad3f4ef982 ci: update version string in docs 2024-05-01 15:35:30 +00:00
dante
d464ddf6b6 chore: medium sized lstm example (#785) 2024-05-01 16:35:11 +01:00
5 changed files with 16 additions and 6 deletions

View File

@@ -1,4 +1,4 @@
ezkl==0.0.0
ezkl==11.0.2
sphinx
sphinx-rtd-theme
sphinxcontrib-napoleon

View File

@@ -1,7 +1,7 @@
import ezkl
project = 'ezkl'
release = '0.0.0'
release = '11.0.2'
version = release

View File

@@ -0,0 +1,9 @@
{
"input_data": [
[
1.514470100402832, 1.519423007965088, 1.5182757377624512,
1.5262789726257324, 1.5298409461975098
]
],
"output_data": [[-0.1862019]]
}

Binary file not shown.

View File

@@ -200,7 +200,7 @@ mod native_tests {
"1l_tiny_div",
];
const TESTS: [&str; 92] = [
const TESTS: [&str; 93] = [
"1l_mlp", //0
"1l_slice",
"1l_concat",
@@ -296,7 +296,8 @@ mod native_tests {
"reducel1",
"reducel2", // 89
"1l_lppool",
"lstm_large", // 91
"lstm_large", // 91
"lstm_medium", // 92
];
const WASM_TESTS: [&str; 46] = [
@@ -535,7 +536,7 @@ mod native_tests {
}
});
seq!(N in 0..=91 {
seq!(N in 0..=92 {
#(#[test_case(TESTS[N])])*
#[ignore]
@@ -623,7 +624,7 @@ mod native_tests {
#(#[test_case(TESTS[N])])*
fn mock_large_batch_public_outputs_(test: &str) {
// currently variable output rank is not supported in ONNX
if test != "gather_nd" && test != "lstm_large" {
if test != "gather_nd" && test != "lstm_large" && test != "lstm_medium" {
crate::native_tests::init_binary();
let test_dir = TempDir::new(test).unwrap();
let path = test_dir.path().to_str().unwrap(); crate::native_tests::mv_test_(path, test);