Staging directives are a dict now.
Now that staging directives are a dict, compute unit description becomes simpler too.