.funcs = &amdgpu_dm_funcs,
 };
 
+
+struct drm_atomic_state *
+dm_atomic_state_alloc(struct drm_device *dev)
+{
+       struct dm_atomic_state *state = kzalloc(sizeof(*state), GFP_KERNEL);
+
+       if (!state || drm_atomic_state_init(dev, &state->base) < 0) {
+               kfree(state);
+               return NULL;
+       }
+
+       return &state->base;
+}
+
+void dm_atomic_state_clear(struct drm_atomic_state *s)
+{
+       struct dm_atomic_state *state = to_dm_atomic_state(s);
+       drm_atomic_state_default_clear(&state->base);
+}
+
+
+static void dm_atomic_state_free(struct drm_atomic_state *state)
+{
+       struct dm_atomic_state *dm_state = to_dm_atomic_state(state);
+
+       drm_atomic_state_default_release(state);
+
+       kfree(dm_state);
+}
+
+
+
 static const struct drm_mode_config_funcs amdgpu_dm_mode_funcs = {
        .fb_create = amdgpu_user_framebuffer_create,
        .output_poll_changed = amdgpu_output_poll_changed,
        .atomic_check = amdgpu_dm_atomic_check,
-       .atomic_commit = drm_atomic_helper_commit
+       .atomic_commit = drm_atomic_helper_commit,
+       .atomic_state_alloc = dm_atomic_state_alloc,
+       .atomic_state_clear = dm_atomic_state_clear,
+       .atomic_state_free = dm_atomic_state_free,
 };
 
 static struct drm_mode_config_helper_funcs amdgpu_dm_mode_config_helperfuncs = {
 
 #define __AMDGPU_DM_TYPES_H__
 
 #include <drm/drmP.h>
+#include <drm/drm_atomic.h>
 
 struct amdgpu_framebuffer;
 struct amdgpu_display_manager;
 
 #define to_dm_crtc_state(x)    container_of(x, struct dm_crtc_state, base)
 
+struct dm_atomic_state {
+       struct drm_atomic_state base;
+};
+
+#define to_dm_atomic_state(x) container_of(x, struct dm_atomic_state, base)
+
+
 /*TODO Jodan Hersen use the one in amdgpu_dm*/
 int amdgpu_dm_plane_init(struct amdgpu_display_manager *dm,
                        struct amdgpu_plane *aplane,