⚡ 修改taichi为packed模式，优化内存占用

MetallyInduced · Jan 1, 2023 · c56ef8b · c56ef8b
1 parent 768735f
commit c56ef8b
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/examples/demag/main_demag.py b/examples/demag/main_demag.py
@@ -16,10 +16,12 @@
 from metalpy.scab.utils.misc import define_inducing_field
 
 from config import get_exec_config
+from metalpy.utils.taichi import ti_prepare
 from metalpy.utils.time import Timer
 
 
 def main(grid_size):
+    ti_prepare(packed=True)
     a, c = 10, 40
     timer = Timer()