@@ -136,2 +126,2 @@
-            SEEN_RTIO_LED = READ_RTIO_LED;
-        };
+        let _ = block_async!(wait_for_virtual_leds_change()).await;
+        info!("switching");

@@ -118,2 +111,2 @@
-        }
-        if READ_RTIO_LED != SEEN_RTIO_LED {
+        if pl::csr::virtual_leds::status_read() != LAST_VIRTUAL_LED_STATUS {
+            LAST_VIRTUAL_LED_STATUS = pl::csr::virtual_leds::status_read();

@@ -673,2 +656,4 @@
     let mut hardware_tick_ts = 0;
 
+    let mut i2c0 = I2c::i2c0();
+    let mut i2c1 = I2c::i2c0();

@@ -126,2 +120,4 @@
 #[cfg(all(feature = "target_kasli_soc", has_drtio))]
 async fn async_rtio_led() {
+    let mut io_expander0 = io_expander::IoExpander::new(unsafe { (&mut i2c::I2C_BUS).as_mut().unwrap() }, 0).unwrap();
+    let mut io_expander1 = io_expander::IoExpander::new(unsafe { (&mut i2c::I2C_BUS).as_mut().unwrap() }, 1).unwrap();

@@ -105,0 +107,4 @@
+
+#[cfg(all(feature = "target_kasli_soc", has_drtio))]
+fn wait_for_virtual_leds_change() -> nb::Result<(), Void> {
+    unsafe {

@@ -105,0 +118,4 @@
+    }
+}
+#[cfg(all(feature = "target_kasli_soc", has_drtio))]
+async fn async_io_expanders_service() {

@@ -105,0 +122,4 @@
+    let mut io_expander0 = io_expander::IoExpander::new(unsafe { (&mut i2c::I2C_BUS).as_mut().unwrap() }, 0).unwrap();
+    let mut io_expander1 = io_expander::IoExpander::new(unsafe { (&mut i2c::I2C_BUS).as_mut().unwrap() }, 1).unwrap();
+    loop {
+        let _ = block_async!(wait_for_virtual_leds_change()).await;

@@ -658,2 +657,4 @@
     let mut hardware_tick_ts = 0;
 
+    let mut io_expander0 = io_expander::IoExpander::new(unsafe { &mut *i2c_ptr }, 0).unwrap();
+    let mut io_expander1 = io_expander::IoExpander::new(unsafe { &mut *i2c_ptr }, 1).unwrap();

@@ -11,2 +13,4 @@
 }
 
+//IO expanders pins
+const IO_DIR_INPUT_ALL: u8 = 0xFF;

@@ -13,0 +22,4 @@
+const IO_DIR_OUT_SFP0_LED: u8 = !0x80;
+
+//IO expander port direction
+const IO_DIR_MAPPING0: [u8; 2] = [

@@ -12,1 +14,4 @@
 
+//IO expanders pins
+const IO_DIR_INPUT_ALL: u8 = 0xFF;
+const IO_DIR_OUT_SFP_TX_DISABLE: u8 = !0x02;

@@ -102,0 +139,4 @@
+            0 => IO_DIR_MAPPING0,
+            1 => IO_DIR_MAPPING1,
+            _ => [IO_DIR_INPUT_ALL; 2],
+        };

@@ -18,2 +40,4 @@
     out_target: [u8; 2],
     registers: Registers,
+    virtual_led_mapping: &'static [(u8, u8, u8)],
+    channel: usize,

@@ -150,5 +174,8 @@ pub fn main_core0() {
 
     task::spawn(report_async_rtio_errors());

@@ -105,0 +120,4 @@
+#[cfg(all(feature = "target_kasli_soc", has_drtio))]
+async fn async_io_expanders_service() {
+    let mut io_expander0 = io_expander::IoExpander::new(unsafe { (&mut i2c::I2C_BUS).as_mut().unwrap() }, 0).unwrap();
+    let mut io_expander1 = io_expander::IoExpander::new(unsafe { (&mut i2c::I2C_BUS).as_mut().unwrap() }, 1).unwrap();

@@ -105,0 +107,4 @@
+#[cfg(all(feature = "target_kasli_soc", has_drtio))]
+static mut LAST_VIRTUAL_LED_STATUS: u8 = 0;
+
+#[cfg(all(feature = "target_kasli_soc", has_drtio))]

@@ -105,0 +114,4 @@
+    io_expander1: RefCell<io_expander::IoExpander>,
+) {
+    loop {
+        let _ = block_async!((|| -> nb::Result<(), Void> {

@@ -470,0 +486,4 @@
+        self.csr_devices.append("virtual_leds")
+
+        self.comb += [self.virtual_leds.get(i).eq(channel.rx_ready)
+                for i, channel in enumerate(self.drtio_transceiver.channels)]