Python number_of_hold_vars示例

示例#1

文件： test_node.py 项目： chengwei920412/jittor-deep_learning

def check(hv, lv, lo):
    import gc
    gc.collect()
    jt.graph_check()
    a, b, c = jt.number_of_hold_vars(), jt.number_of_lived_vars(), jt.number_of_lived_ops()
    assert (a,b,c)==(hv,lv,lo), (a, b, c, jt.dump_all_graphs().nodes_info)

示例#2

文件： test_fused_op.py 项目： chengwei920412/jittor-deep_learning

 def check(hv, lv, lo):
     self.assertEqual(jt.number_of_hold_vars(), hv)
     self.assertEqual(jt.number_of_lived_vars(), lv)
     self.assertEqual(jt.number_of_lived_ops(), lo)