if (param_size > 8)
                return -EINVAL;
 
+       pm_runtime_get_sync(hba->dev);
        ret = ufshcd_read_desc_param(hba, desc_id, desc_index,
                                param_offset, desc_buf, param_size);
+       pm_runtime_put_sync(hba->dev);
        if (ret)
                return -EINVAL;
        switch (param_size) {
        desc_buf = kzalloc(QUERY_DESC_MAX_SIZE, GFP_ATOMIC);            \
        if (!desc_buf)                                                  \
                return -ENOMEM;                                         \
+       pm_runtime_get_sync(hba->dev);                                  \
        ret = ufshcd_query_descriptor_retry(hba,                        \
                UPIU_QUERY_OPCODE_READ_DESC, QUERY_DESC_IDN_DEVICE,     \
                0, 0, desc_buf, &desc_len);                             \
                goto out;                                               \
        ret = snprintf(buf, PAGE_SIZE, "%s\n", desc_buf);               \
 out:                                                                   \
+       pm_runtime_put_sync(hba->dev);                                  \
        kfree(desc_buf);                                                \
        return ret;                                                     \
 }                                                                      \
        struct device_attribute *attr, char *buf)                       \
 {                                                                      \
        bool flag;                                                      \
+       int ret;                                                        \
        struct ufs_hba *hba = dev_get_drvdata(dev);                     \
-       if (ufshcd_query_flag(hba, UPIU_QUERY_OPCODE_READ_FLAG,         \
-               QUERY_FLAG_IDN##_uname, &flag))                         \
+       pm_runtime_get_sync(hba->dev);                                  \
+       ret = ufshcd_query_flag(hba, UPIU_QUERY_OPCODE_READ_FLAG,       \
+               QUERY_FLAG_IDN##_uname, &flag);                         \
+       pm_runtime_put_sync(hba->dev);                                  \
+       if (ret)                                                        \
                return -EINVAL;                                         \
        return sprintf(buf, "%s\n", flag ? "true" : "false");           \
 }                                                                      \
 {                                                                      \
        struct ufs_hba *hba = dev_get_drvdata(dev);                     \
        u32 value;                                                      \
-       if (ufshcd_query_attr(hba, UPIU_QUERY_OPCODE_READ_ATTR,         \
-               QUERY_ATTR_IDN##_uname, 0, 0, &value))                  \
+       int ret;                                                        \
+       pm_runtime_get_sync(hba->dev);                                  \
+       ret = ufshcd_query_attr(hba, UPIU_QUERY_OPCODE_READ_ATTR,       \
+               QUERY_ATTR_IDN##_uname, 0, 0, &value);                  \
+       pm_runtime_put_sync(hba->dev);                                  \
+       if (ret)                                                        \
                return -EINVAL;                                         \
        return sprintf(buf, "0x%08X\n", value);                         \
 }                                                                      \
        struct scsi_device *sdev = to_scsi_device(dev);
        struct ufs_hba *hba = shost_priv(sdev->host);
        u8 lun = ufshcd_scsi_to_upiu_lun(sdev->lun);
+       int ret;
 
-       if (ufshcd_query_attr(hba, UPIU_QUERY_OPCODE_READ_ATTR,
-               QUERY_ATTR_IDN_DYN_CAP_NEEDED, lun, 0, &value))
+       pm_runtime_get_sync(hba->dev);
+       ret = ufshcd_query_attr(hba, UPIU_QUERY_OPCODE_READ_ATTR,
+               QUERY_ATTR_IDN_DYN_CAP_NEEDED, lun, 0, &value);
+       pm_runtime_put_sync(hba->dev);
+       if (ret)
                return -EINVAL;
        return sprintf(buf, "0x%08X\n", value);
 }