fix several test imports, add Elaboratable

[soc.git] / src / TLB / ariane / tlb_content.py

diff --git a/src/TLB/ariane/tlb_content.py b/src/TLB/ariane/tlb_content.py
index 024c5697e1f96c77cea6058a9c2785a501c058cd..5c005b84bbf97d98b6879f71d4adbb6ed888aa61 100644 (file)
--- a/src/TLB/ariane/tlb_content.py
+++ b/src/TLB/ariane/tlb_content.py
@@ -1,6 +1,7 @@
-from nmigen import Signal, Module, Cat, Const
+from nmigen import Signal, Module, Cat, Const, Elaboratable
+
+from TLB.ariane.ptw import TLBUpdate, PTE
 
-from ptw import TLBUpdate, PTE
 
 class TLBEntry:
     def __init__(self, asid_width):
@@ -23,7 +24,7 @@ class TLBEntry:
         return [self.asid, self.vpn0, self.vpn1, self.vpn2,
                 self.is_2M, self.is_1G, self.valid]
 
-class TLBContent:
+class TLBContent(Elaboratable):
     def __init__(self, pte_width, asid_width):
         self.asid_width = asid_width
         self.pte_width = pte_width