tools/litex_server: fix comms import

diff --git a/litex/tools/litex_server.py b/litex/tools/litex_server.py
index 1dfd2b2e953e864a33cc987e2204cf4ddebcd063..c2c2f98e34fc022916f7c3b5ca6bc85ab5314a0e 100755 (executable)
--- a/litex/tools/litex_server.py
+++ b/litex/tools/litex_server.py
@@ -141,7 +141,7 @@ def main():
 
 
     if args.uart:
-        from litex.tools.remote import CommUART
+        from litex.tools.remote.comm_uart import CommUART
         if args.uart_port is None:
             print("Need to specify --uart-port, exiting.")
             exit()
@@ -150,13 +150,13 @@ def main():
         print("[CommUART] port: {} / baudrate: {} / ".format(uart_port, uart_baudrate), end="")
         comm = CommUART(uart_port, uart_baudrate)
     elif args.udp:
-        from litex.tools.remote import CommUDP
+        from litex.tools.remote.comm_udp import CommUDP
         udp_ip = args.udp_ip
         udp_port = int(args.udp_port)
         print("[CommUDP] ip: {} / port: {} / ".format(udp_ip, udp_port), end="")
         comm = CommUDP(udp_ip, udp_port)
     elif args.pcie:
-        from litex.tools.remote import CommPCIe
+        from litex.tools.remote.comm_pcie import CommPCIe
         pcie_bar = args.pcie_bar
         if args.pcie_bar is None:
             print("Need to speficy --pcie-bar, exiting.")
@@ -164,7 +164,7 @@ def main():
         print("[CommPCIe] bar: {} / ".format(args.pcie_bar), end="")
         comm = CommPCIe(args.pcie_bar)
     elif args.usb:
-        from litex.tools.remote import CommUSB
+        from litex.tools.remote.comm_usb import CommUSB
         if args.usb_pid is None and args.usb_vid is None:
             print("Need to speficy --usb-vid or --usb-pid, exiting.")
             exit()