diff --git a/libgralloc/alloc_controller.cpp b/libgralloc/alloc_controller.cpp
index 4daadb6..5a6fe6e 100644
--- a/libgralloc/alloc_controller.cpp
+++ b/libgralloc/alloc_controller.cpp
@@ -104,7 +104,7 @@
 {
     int stride = ALIGN(width, 32);
     // Currently surface padding is only computed for RGB* surfaces.
-    if (format < 0x7) {
+    if (format <= HAL_PIXEL_FORMAT_sRGB_888) {
         // Don't add any additional padding if debug.gralloc.map_fb_memory
         // is enabled
         char property[PROPERTY_VALUE_MAX];
@@ -118,6 +118,7 @@
         switch(format)
         {
             case HAL_PIXEL_FORMAT_RGB_888:
+            case HAL_PIXEL_FORMAT_sRGB_888:
                 bpp = 3;
                 break;
             case HAL_PIXEL_FORMAT_RGB_565:
@@ -280,9 +281,11 @@
         case HAL_PIXEL_FORMAT_RGBA_8888:
         case HAL_PIXEL_FORMAT_RGBX_8888:
         case HAL_PIXEL_FORMAT_BGRA_8888:
+        case HAL_PIXEL_FORMAT_sRGB_A_8888:
             size = alignedw * alignedh * 4;
             break;
         case HAL_PIXEL_FORMAT_RGB_888:
+        case HAL_PIXEL_FORMAT_sRGB_888:
             size = alignedw * alignedh * 3;
             break;
         case HAL_PIXEL_FORMAT_RGB_565:
diff --git a/libgralloc/gpu.cpp b/libgralloc/gpu.cpp
index aacb18c..2b123dd 100644
--- a/libgralloc/gpu.cpp
+++ b/libgralloc/gpu.cpp
@@ -173,7 +173,7 @@
 {
     *bufferType = BUFFER_TYPE_VIDEO;
 
-    if (inputFormat < 0x7) {
+    if (inputFormat <= HAL_PIXEL_FORMAT_sRGB_888) {
         // RGB formats
         *bufferType = BUFFER_TYPE_UI;
     } else if ((inputFormat == HAL_PIXEL_FORMAT_R_8) ||