add initial compute_v2 support