unsigned size = vma->node.size << PAGE_SHIFT;
 
        /* Print a message if we try to purge a vma in use */
-       if (WARN_ON(vma->inuse > 0))
+       if (GEM_WARN_ON(vma->inuse > 0))
                return;
 
        /* Don't do anything if the memory isn't mapped */
 void msm_gem_unmap_vma(struct msm_gem_address_space *aspace,
                struct msm_gem_vma *vma)
 {
-       if (!WARN_ON(!vma->iova))
+       if (!GEM_WARN_ON(!vma->iova))
                vma->inuse--;
 }
 
        unsigned size = npages << PAGE_SHIFT;
        int ret = 0;
 
-       if (WARN_ON(!vma->iova))
+       if (GEM_WARN_ON(!vma->iova))
                return -EINVAL;
 
        /* Increase the usage counter */
 void msm_gem_close_vma(struct msm_gem_address_space *aspace,
                struct msm_gem_vma *vma)
 {
-       if (WARN_ON(vma->inuse > 0 || vma->mapped))
+       if (GEM_WARN_ON(vma->inuse > 0 || vma->mapped))
                return;
 
        spin_lock(&aspace->lock);
 {
        int ret;
 
-       if (WARN_ON(vma->iova))
+       if (GEM_WARN_ON(vma->iova))
                return -EBUSY;
 
        spin_lock(&aspace->lock);