add balance fence

vyperlang · Jun 4, 2024 · 2dad58b · 2dad58b
1 parent 7065892
commit 2dad58b
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/vyper/venom/passes/dft.py b/vyper/venom/passes/dft.py
@@ -8,7 +8,7 @@
 from vyper.venom.function import IRFunction
 from vyper.venom.passes.base_pass import IRPass
 
-_ALL = ("storage", "transient", "memory", "immutables")
+_ALL = ("storage", "transient", "memory", "immutables", "balance")
 
 writes = {
     "sstore": "storage",
@@ -37,6 +37,8 @@
     "call": _ALL,
     "delegatecall": _ALL,
     "staticcall": _ALL,
+    "balance": "balance",
+    "selfbalance": "balance",
     "log": "memory",
     "revert": "memory",
     "return": "memory",