A faster way to estimate AI power consumption