[aot] Fix 140 convention to match master (#8306)

Issue: #

### Brief Summary

<!--
copilot:summary
-->
### <samp>🤖 Generated by Copilot at 184fd66</samp>

Update `dr.py` to use the new JSON format for AOT compilation. This
allows the script to generate correct device runtime code for kernels
and buffers.

### Walkthrough

<!--
copilot:walkthrough
-->
### <samp>🤖 Generated by Copilot at 184fd66</samp>

* Adapt the `ArgumentAttributes` and `Buffer` classes to the new JSON
format of the kernel arguments and buffer root IDs
([link](https://github.com/taichi-dev/taichi/pull/8306/files?diff=unified&w=0#diff-cbaba00c0691d883383ed429835dfde47b0b5c545b5990c3c75d47a8d63c1716L33-R45),
[link](https://github.com/taichi-dev/taichi/pull/8306/files?diff=unified&w=0#diff-cbaba00c0691d883383ed429835dfde47b0b5c545b5990c3c75d47a8d63c1716L78-R79))
* Adapt the array access list to the new JSON format of the access modes
([link](https://github.com/taichi-dev/taichi/pull/8306/files?diff=unified&w=0#diff-cbaba00c0691d883383ed429835dfde47b0b5c545b5990c3c75d47a8d63c1716L69-R71))

python/taichi/aot/conventions/gfxruntime140/dr.py

@@ -30,19 +30,19 @@ def __init__(self, j: Dict[str, Any]) -> None:
 @json_data_model
 class ArgumentAttributes:
     def __init__(self, j: Dict[str, Any]) -> None:
-        dtype = j["dtype"]
-        element_shape = j["element_shape"]
-        field_dim = j["field_dim"]
-        fmt = j["format"]
-        index = j["index"]
-        is_array = j["is_array"]
-        offset_in_mem = j["offset_in_mem"]
-        stride = j["stride"]
+        index = j["key"][0]
+        dtype = j["value"]["dtype"]
+        element_shape = j["value"]["element_shape"]
+        field_dim = j["value"]["field_dim"]
+        fmt = j["value"]["format"]
+        is_array = j["value"]["is_array"]
+        offset_in_mem = j["value"]["offset_in_mem"]
+        stride = j["value"]["stride"]
         # (penguinliong) Note that the name field is optional for kernels.
         # Kernels are always launched by indexed arguments and this is for
         # debugging and header generation only.
-        name = j["name"] if "name" in j and len(j["name"]) > 0 else None
-        ptype = j["ptype"] if "ptype" in j else None
+        name = j["value"]["name"] if "name" in j["value"] and len(j["value"]["name"]) > 0 else None
+        ptype = j["value"]["ptype"] if "ptype" in j["value"] else None
 
         self.dtype: int = int(dtype)
         self.element_shape: List[int] = [int(x) for x in element_shape]
@@ -66,16 +66,17 @@ def __init__(self, j: Dict[str, Any]) -> None:
         rets_bytes_ = j["rets_bytes_"]
 
         self.arg_attribs_vec_: List[ArgumentAttributes] = [ArgumentAttributes(x) for x in arg_attribs_vec_]
+        self.arg_attribs_vec_.sort(key=lambda x: x.index)
         self.args_bytes_: int = int(args_bytes_)
-        self.arr_access: List[int] = [int(x) for x in arr_access]
+        self.arr_access: List[int] = [int(x["value"]) for x in arr_access]
         self.ret_attribs_vec_: List[ArgumentAttributes] = [ArgumentAttributes(x) for x in ret_attribs_vec_]
         self.rets_bytes_: int = int(rets_bytes_)
 
 
 @json_data_model
 class Buffer:
     def __init__(self, j: Dict[str, Any]) -> None:
-        root_id = j["root_id"]
+        root_id = j["root_id"][0]
         ty = j["type"]
 
         self.root_id: int = int(root_id)