Enable RANGE definition in shader YAML for repetitive shader generation (#4196)

Summary:
Pull Request resolved: #4196

GPUInfo requires the compilation of a ton of shaders with small variations to emulate OpenCL's kernel compilation on runtime with strings. This diff enables a RANGE implementation to generate these shaders dynamically without typing each value out.

The format is

    - RANGE: [start, end] // Inclusive
      SUFFIX: string // Optional

If a suffix is selected, the shaders will be called `name_suffix_K	` for each value K from the range. If no suffix is selected, the shaders will be simply called `name_K	`

Reviewed By: SS-JIA

Differential Revision: D59588223

fbshipit-source-id: 67394c6b860c09e324baa45ccf7f5b7d6d555ed2

Author: Esteban Padilla Cerdio

backends/vulkan/runtime/gen_vulkan_spv.py

@@ -525,8 +525,29 @@ def generateVariantCombinations(
             if param_name not in exclude_params:
                 param_values = []
                 for value in value_list:
-                    suffix = value.get("SUFFIX", value["VALUE"])
-                    param_values.append((param_name, suffix, value["VALUE"]))
+                    if "RANGE" in value:
+                        value_range = value["RANGE"]
+                        suffix = value.get("SUFFIX", "")
+                        if isinstance(value_range, list) and len(value_range) == 2:
+                            for i in range(value_range[0], value_range[1] + 1):
+                                curr_suffix = (
+                                    suffix + "_" + str(i) if suffix else str(i)
+                                )
+                                param_values.append((param_name, curr_suffix, str(i)))
+                        else:
+                            raise ValueError(
+                                f"{value['RANGE']} is not a valid range. Must be in format [start, end] (inclusive)."
+                            )
+
+                    elif "VALUE" in value:
+                        suffix = value.get("SUFFIX", value["VALUE"])
+                        param_values.append((param_name, suffix, value["VALUE"]))
+
+                    else:
+                        raise KeyError(
+                            "Parameter must be 'VALUE: string' or 'RANGE: [a, b]'"
+                        )
+
                 all_iterated_params.append(param_values)
 
         return list(product(*all_iterated_params))