Translation (#17)

* finalize prompts

* translation scripts

* gpt translate outputs

* add translation outputs

* translation results

Author: Dando18

drivers/problem-sizes.json

@@ -4,7 +4,7 @@
     "omp": "(1<<10)",
     "mpi": "(1<<10)",
     "mpi+omp": "(1<<10)",
-    "kokkos": "(1<<10)",
+    "kokkos": "(1<<9)",
     "cuda": "(1<<10)",
     "hip": "(1<<10)"
   },
@@ -18,13 +18,13 @@
     "hip": "(1<<10)"
   },
   "02_dense_la_gemm": {
-    "serial": "(1<<10)",
-    "omp": "(1<<10)",
-    "mpi": "(1<<10)",
-    "mpi+omp": "(1<<10)",
-    "kokkos": "(1<<10)",
-    "cuda": "(1<<10)",
-    "hip": "(1<<10)"
+    "serial": "(1<<9)",
+    "omp": "(1<<9)",
+    "mpi": "(1<<9)",
+    "mpi+omp": "(1<<9)",
+    "kokkos": "(1<<9)",
+    "cuda": "(1<<9)",
+    "hip": "(1<<9)"
   },
   "03_dense_la_axpy": {
     "serial": "(1<<24)",
@@ -47,8 +47,8 @@
   "05_fft_inverse_fft": {
     "serial": "(1<<17)",
     "omp": "(1<<17)",
-    "mpi": "(1<<18)",
-    "mpi+omp": "(1<<18)",
+    "mpi": "(1<<17)",
+    "mpi+omp": "(1<<17)",
     "kokkos": "(1<<17)",
     "cuda": "(1<<17)",
     "hip": "(1<<17)"
@@ -72,22 +72,22 @@
     "hip": "(1<<18)"
   },
   "08_fft_split_fft": {
-    "serial": "(1<<18)",
-    "omp": "(1<<18)",
-    "mpi": "(1<<19)",
-    "mpi+omp": "(1<<19)",
-    "kokkos": "(1<<18)",
-    "cuda": "(1<<18)",
-    "hip": "(1<<18)"
+    "serial": "(1<<17)",
+    "omp": "(1<<17)",
+    "mpi": "(1<<17)",
+    "mpi+omp": "(1<<17)",
+    "kokkos": "(1<<17)",
+    "cuda": "(1<<17)",
+    "hip": "(1<<17)"
   },
   "09_fft_fft_out_of_place": {
-    "serial": "(1<<18)",
-    "omp": "(1<<18)",
-    "mpi": "(1<<19)",
-    "mpi+omp": "(1<<19)",
-    "kokkos": "(1<<18)",
-    "cuda": "(1<<18)",
-    "hip": "(1<<18)"
+    "serial": "(1<<17)",
+    "omp": "(1<<17)",
+    "mpi": "(1<<17)",
+    "mpi+omp": "(1<<17)",
+    "kokkos": "(1<<17)",
+    "cuda": "(1<<17)",
+    "hip": "(1<<17)"
   },
   "10_geometry_convex_hull": {
     "serial": "(1<<14)",
@@ -109,28 +109,28 @@
   },
   "12_geometry_smallest_triangle": {
     "serial": "(1<<8)",
-    "omp": "(1<<8)",
+    "omp": "(1<<7)",
     "mpi": "(1<<8)",
     "mpi+omp": "(1<<8)",
-    "kokkos": "(1<<8)",
+    "kokkos": "(1<<7)",
     "cuda": "(1<<10)",
     "hip": "(1<<10)"
   },
   "13_geometry_closest_pair_2d": {
     "serial": "(1<<14)",
-    "omp": "(1<<14)",
+    "omp": "(1<<13)",
     "mpi": "(1<<14)",
     "mpi+omp": "(1<<14)",
-    "kokkos": "(1<<14)",
+    "kokkos": "(1<<13)",
     "cuda": "(1<<14)",
     "hip": "(1<<14)"
   },
   "14_geometry_closest_pair_1d": {
     "serial": "(1<<14)",
-    "omp": "(1<<14)",
+    "omp": "(1<<13)",
     "mpi": "(1<<14)",
     "mpi+omp": "(1<<14)",
-    "kokkos": "(1<<14)",
+    "kokkos": "(1<<13)",
     "cuda": "(1<<14)",
     "hip": "(1<<14)"
   },
@@ -153,13 +153,13 @@
     "hip": "(1<<12)"
   },
   "17_graph_highest_degree": {
-    "serial": "(1<<13)",
-    "omp": "(1<<13)",
-    "mpi": "(1<<13)",
-    "mpi+omp": "(1<<13)",
+    "serial": "(1<<12)",
+    "omp": "(1<<12)",
+    "mpi": "(1<<12)",
+    "mpi+omp": "(1<<12)",
     "kokkos": "(1<<12)",
-    "cuda": "(1<<13)",
-    "hip": "(1<<13)"
+    "cuda": "(1<<12)",
+    "hip": "(1<<12)"
   },
   "18_graph_count_components": {
     "serial": "(1<<12)",

generate/generate-openai.py

@@ -100,7 +100,7 @@ def get_max_requests_per_day(model: str) -> Optional[int]:
 def postprocess(prompt: str, output: str) -> str:
     """ Postprocess the output. """
     # remove leading ```, ```cpp, and trailing ```
-    output = output.lstrip("```cpp").lstrip("```").rstrip("```")
+    output = output.strip().lstrip("```cpp").lstrip("```").rstrip("```")
 
     # remove prompt if it included it
     if output.startswith(prompt):