add comment

7 years ago · c88f58dbd8
parent 3b44b849d3
commit c88f58dbd8
1 changed files with 1 additions and 0 deletions
--- 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 */