Python number_of_lived_ops Examples

Example #1

File: test_node.py Project: 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)

Example #2

File: test_fused_op.py Project: 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)