Consolidate flash algorithm init

2 files changed, 3 insertions, 5 deletions

diff --git a/imxrt1040evk/src/lib.rs b/imxrt1040evk/src/lib.rs
index ad2bb57..ebc7102 100644
--- a/imxrt1040evk/src/lib.rs
+++ b/imxrt1040evk/src/lib.rs
@@ -2,6 +2,4 @@
 
 use cortex_m as _;
 
-pub use imxrt1040_flash_algos::flash::winbond::W25q64 as Flash;
-
-pub type Algorithm = imxrt1040_flash_algos::Algorithm<Flash>;
+pub type Algorithm = imxrt1040_flash_algos::Algorithm<{ 128 / 8 * 1024 * 1024 }>;
diff --git a/imxrt1040evk/src/main.rs b/imxrt1040evk/src/main.rs
index f144213..3b71188 100644
--- a/imxrt1040evk/src/main.rs
+++ b/imxrt1040evk/src/main.rs
@@ -13,12 +13,12 @@ flash_algorithm::algorithm!(Algorithm, {
     // We support page crossings. Suggest a larger page
     // size so that probe-rs places larger buffers into
     // RAM per programming operation, reducing overhead.
-    page_size: 4096,
+    page_size: Algorithm::sector_size_bytes() as _,
     empty_value: 0xFF,
     program_time_out: 1000,
     erase_time_out: 2000,
     sectors: [{
-        size: Algorithm::sector_size_bytes() as u32,
+        size: Algorithm::sector_size_bytes() as _,
         address: 0x0,
     }]
 });