Adding extension support for compute