[examples] Update GPT-3

Author: gilbertmike

examples/workloads/gpt3_175B.yaml

@@ -33,24 +33,26 @@ workload:
   persistent_tensors: weight - Intermediates
 
   einsums:
-  - einsum: I(output)[b, m, d] = I_in(input)[b, m, d]
+  - einsum: "I[b, m, d] = I_in[b, m, d]"
     is_copy_operation: True
-  - V[b, m, h, e] = I[b, m, d] * WV[h, e, d]
-  - K[b, m, h, e] = I[b, m, d] * WK[h, e, d]
-  - Q[b, m, h, e] = I[b, m, d] * WQ[h, e, d]
-  - QK[b, m, p, h] = Q(input)[b, m, h, e] * K(weight)[b, M=p, h, e]
-  - QK_softmax[b, m, p, h] = QK(input)[b, m, p, h]
-  - AV[b, m, h, f] = QK_softmax(input)[b, m, p, h] * V(weight)[b, M=p, H=h, E=f]
-  - Z[b, m, g] = AV[b, m, h, f] * WZ[h, f, g]
-  - FFA[b, m, c] = Z[b, m, g] * WFFA[g, c]
-  - FFB[b, m, j] = FFA[b, m, c] * WFFB[c, j]
+  - "V[b, m, h, e] = I[b, m, d] * WV[h, e, d]"
+  - "K[b, m, h, e] = I[b, m, d] * WK[h, e, d]"
+  - "Q[b, m, h, e] = I[b, m, d] * WQ[h, e, d]"
+  - einsum: "QK[b, m, p, h] = Q[b, m, h, e] * K[b, M: p, h, e]"
+    renames: {input: Q}
+  - "QK_softmax[b, m, p, h] = QK[b, m, p, h]"
+  - einsum: "AV[b, m, h, f] = QK_softmax[b, m, p, h] * V[b, M: p, h, E: f]"
+    renames: {input: QK_softmax}
+  - "Z[b, m, g] = AV[b, m, h, f] * WZ[h, f, g]"
+  - "FFA[b, m, c] = Z[b, m, g] * WFFA[g, c]"
+  - "FFB[b, m, j] = FFA[b, m, c] * WFFB[c, j]"
 
 renames:
   einsums:
   - name: default
     tensor_accesses:
     - name: input
-      source: Inputs & Intermediates
+      source: Inputs & Intermediates if len(All) == 3 else Inputs
       expected_count: 1
     - name: output
       source: Outputs