diff --git a/src/amd/amdgpu/atom.c b/src/amd/amdgpu/atom.c
index 5f854ed..9a04857 100644
--- a/src/amd/amdgpu/atom.c
+++ b/src/amd/amdgpu/atom.c
@@ -22,14 +22,16 @@
  * Author: Stanislaw Skowronek
  */
 
+#include "common.h"
+
 #include <linux/module.h>
 #include <linux/sched.h>
 #include <linux/slab.h>
 #include <linux/delay.h>
 #include <asm/unaligned.h>
 
-#include <drm/drm_util.h>
-#include <drm/drm_print.h>
+//#include <drm/drm_util.h>
+//#include <drm/drm_print.h>
 
 #define ATOM_DEBUG
 
diff --git a/src/amd/common.h b/src/amd/common.h
index b999809..5e32fcd 100644
--- a/src/amd/common.h
+++ b/src/amd/common.h
@@ -24,10 +24,16 @@ Place, Suite 330, Boston, MA 02111-1307 USA
 #include "vendor-reset-dev.h"
 
 /* these are to remove the dependency on DRM */
+#include <linux/kgdb.h>
 #define DRM_INFO(fmt, args...) pr_info("vendor-reset-drm: " fmt, ##args)
 #define DRM_ERROR(fmt, args...) pr_err("vendor-reset-drm: " fmt, ##args)
 #define DRM_DEBUG(fmt, args...) pr_debug("vendor-reset-drm: " fmt, ##args)
-
+static inline bool drm_can_sleep(void)
+{
+	if (in_atomic() || in_dbg_master() || irqs_disabled())
+		return false;
+	return true;
+}
 
 #define RREG32(reg)                                                          \
   ({                                                                         \