diff --git a/paddle/fluid/platform/device_context.h b/paddle/fluid/platform/device_context.h index 500891ac7aee83549ead773aea7f95d978d00749..603b890af13b529c490c29112a73a09cc815d07a 100644 --- a/paddle/fluid/platform/device_context.h +++ b/paddle/fluid/platform/device_context.h @@ -79,6 +79,7 @@ class CUDADeviceContext : public DeviceContext { /*! \brief Return place in the device context. */ Place GetPlace() const override; + /*! \brief Return compute capability in the device context. */ int GetComputeCapability() const; /*! \brief Return the max physical thread count in the device context */