diff --git a/drivers/gpu/nvgpu/include/nvgpu/gr/fs_state.h b/drivers/gpu/nvgpu/include/nvgpu/gr/fs_state.h
index 3212c6b7b..4e07884ea 100644
--- a/drivers/gpu/nvgpu/include/nvgpu/gr/fs_state.h
+++ b/drivers/gpu/nvgpu/include/nvgpu/gr/fs_state.h
@@ -23,9 +23,33 @@
 #ifndef NVGPU_GR_FS_STATE_H
 #define NVGPU_GR_FS_STATE_H
 
+/**
+ * @file
+ *
+ * common.gr.fs_state unit interface
+ */
 struct gk20a;
 struct nvgpu_gr_config;
 
+/**
+ * @brief Initialize GR engine h/w state post-floorsweeping.
+ *
+ * @param g[in]			Pointer to GPU driver struct.
+ * @param config[in]		Pointer to GR config struct.
+ *
+ * This function initializes GR engine h/w state after considering
+ * floorsweeping.
+ *
+ * It is possible that certain TPC (and hence SMs) in GPC are
+ * floorswept and hence not available for any processing. In this case
+ * common.gr unit is responsible to enumerate only available TPCs
+ * and configure GR engine h/w registers with available GPC/TPC/SM count
+ * and mapping.
+ *
+ * @return 0 in case of success, < 0 in case of failure.
+ * @retval -ENOMEM if memory allocation fails for any internal data
+ *         structure.
+ */
 int nvgpu_gr_fs_state_init(struct gk20a *g, struct nvgpu_gr_config *config);
 
 #endif /* NVGPU_GR_FS_STATE_H */
diff --git a/drivers/gpu/nvgpu/include/nvgpu/gr/gr.h b/drivers/gpu/nvgpu/include/nvgpu/gr/gr.h
index 19c3c17bf..32886242d 100644
--- a/drivers/gpu/nvgpu/include/nvgpu/gr/gr.h
+++ b/drivers/gpu/nvgpu/include/nvgpu/gr/gr.h
@@ -74,6 +74,8 @@
  * Static Design
  * =============
  *
+ *   + include/nvgpu/gr/fs_state.h
+ *
  * Resource utilization
  * --------------------
  *