tgsi_to_nir: add basic compute shader support