Add ComputeCapability argument