diff --git a/gpu_bdb/queries/q01/gpu_bdb_query_01_dask_sql.py b/gpu_bdb/queries/q01/gpu_bdb_query_01_dask_sql.py index 0bc58976..10fedfc8 100755 --- a/gpu_bdb/queries/q01/gpu_bdb_query_01_dask_sql.py +++ b/gpu_bdb/queries/q01/gpu_bdb_query_01_dask_sql.py @@ -14,6 +14,7 @@ # limitations under the License. # +from nvtx import annotate from bdb_tools.cluster_startup import attach_to_cluster from bdb_tools.utils import ( @@ -70,8 +71,11 @@ def main(data_dir, client, c, config): c.drop_table("distinct_table") return result - -if __name__ == "__main__": +@annotate("QUERY1", color="green", domain="gpu-bdb") +def start_run(): config = gpubdb_argparser() client, c = attach_to_cluster(config, create_sql_context=True) run_query(config=config, client=client, query_func=main, sql_context=c) + +if __name__ == "__main__": + start_run() diff --git a/gpu_bdb/queries/q02/gpu_bdb_query_02_dask_sql.py b/gpu_bdb/queries/q02/gpu_bdb_query_02_dask_sql.py index 6c55c9e0..1d6c17a2 100755 --- a/gpu_bdb/queries/q02/gpu_bdb_query_02_dask_sql.py +++ b/gpu_bdb/queries/q02/gpu_bdb_query_02_dask_sql.py @@ -14,6 +14,7 @@ # limitations under the License. # +from nvtx import annotate from bdb_tools.cluster_startup import attach_to_cluster from bdb_tools.utils import ( @@ -82,7 +83,11 @@ def main(data_dir, client, c, config): return result -if __name__ == "__main__": +@annotate("QUERY2", color="green", domain="gpu-bdb") +def start_run(): config = gpubdb_argparser() client, c = attach_to_cluster(config, create_sql_context=True) run_query(config=config, client=client, query_func=main, sql_context=c) + +if __name__ == "__main__": + start_run() diff --git a/gpu_bdb/queries/q03/gpu_bdb_query_03_dask_sql.py b/gpu_bdb/queries/q03/gpu_bdb_query_03_dask_sql.py index e9a82876..c89bc50f 100755 --- a/gpu_bdb/queries/q03/gpu_bdb_query_03_dask_sql.py +++ b/gpu_bdb/queries/q03/gpu_bdb_query_03_dask_sql.py @@ -14,6 +14,7 @@ # limitations under the License. # +from nvtx import annotate from bdb_tools.cluster_startup import attach_to_cluster from bdb_tools.utils import ( @@ -95,7 +96,11 @@ def main(data_dir, client, c, config): return result -if __name__ == "__main__": +@annotate("QUERY3", color="green", domain="gpu-bdb") +def start_run(): config = gpubdb_argparser() client, c = attach_to_cluster(config, create_sql_context=True) run_query(config=config, client=client, query_func=main, sql_context=c) + +if __name__ == "__main__": + start_run() diff --git a/gpu_bdb/queries/q04/gpu_bdb_query_04_dask_sql.py b/gpu_bdb/queries/q04/gpu_bdb_query_04_dask_sql.py index 89fddc90..d41e2827 100755 --- a/gpu_bdb/queries/q04/gpu_bdb_query_04_dask_sql.py +++ b/gpu_bdb/queries/q04/gpu_bdb_query_04_dask_sql.py @@ -14,6 +14,7 @@ # limitations under the License. # +from nvtx import annotate import cudf from bdb_tools.cluster_startup import attach_to_cluster @@ -88,7 +89,11 @@ def main(data_dir, client, c, config): return result_df -if __name__ == "__main__": +@annotate("QUERY4", color="green", domain="gpu-bdb") +def start_run(): config = gpubdb_argparser() client, c = attach_to_cluster(config, create_sql_context=True) run_query(config=config, client=client, query_func=main, sql_context=c) + +if __name__ == "__main__": + start_run() diff --git a/gpu_bdb/queries/q05/gpu_bdb_query_05_dask_sql.py b/gpu_bdb/queries/q05/gpu_bdb_query_05_dask_sql.py index 6ec6b4c5..28ffc8d4 100755 --- a/gpu_bdb/queries/q05/gpu_bdb_query_05_dask_sql.py +++ b/gpu_bdb/queries/q05/gpu_bdb_query_05_dask_sql.py @@ -14,6 +14,7 @@ # limitations under the License. # +from nvtx import annotate from bdb_tools.cluster_startup import attach_to_cluster from dask.distributed import wait from dask import delayed @@ -91,7 +92,11 @@ def main(data_dir, client, c, config): return results_dict -if __name__ == "__main__": +@annotate("QUERY5", color="green", domain="gpu-bdb") +def start_run(): config = gpubdb_argparser() client, c = attach_to_cluster(config, create_sql_context=True) run_query(config=config, client=client, query_func=main, sql_context=c) + +if __name__ == "__main__": + start_run() diff --git a/gpu_bdb/queries/q06/gpu_bdb_query_06_dask_sql.py b/gpu_bdb/queries/q06/gpu_bdb_query_06_dask_sql.py index 2d1a6a6f..319887f4 100755 --- a/gpu_bdb/queries/q06/gpu_bdb_query_06_dask_sql.py +++ b/gpu_bdb/queries/q06/gpu_bdb_query_06_dask_sql.py @@ -14,6 +14,7 @@ # limitations under the License. # +from nvtx import annotate from bdb_tools.cluster_startup import attach_to_cluster from bdb_tools.utils import ( @@ -96,8 +97,11 @@ def main(data_dir, client, c, config): return result -if __name__ == "__main__": +@annotate("QUERY6", color="green", domain="gpu-bdb") +def start_run(): config = gpubdb_argparser() client, c = attach_to_cluster(config, create_sql_context=True) run_query(config=config, client=client, query_func=main, sql_context=c) +if __name__ == "__main__": + start_run() diff --git a/gpu_bdb/queries/q07/gpu_bdb_query_07_dask_sql.py b/gpu_bdb/queries/q07/gpu_bdb_query_07_dask_sql.py index b18a554f..48c8070c 100755 --- a/gpu_bdb/queries/q07/gpu_bdb_query_07_dask_sql.py +++ b/gpu_bdb/queries/q07/gpu_bdb_query_07_dask_sql.py @@ -14,6 +14,7 @@ # limitations under the License. # +from nvtx import annotate from bdb_tools.cluster_startup import attach_to_cluster from bdb_tools.utils import ( @@ -68,7 +69,11 @@ def main(data_dir, client, c, config): return result +@annotate("QUERY7", color="green", domain="gpu-bdb") +def start_run(): + config = gpubdb_argparser() + client, c = attach_to_cluster(config, create_sql_context=True) + run_query(config=config, client=client, query_func=main, sql_context=c) + if __name__ == "__main__": - config = gpubdb_argparser() - client, c = attach_to_cluster(config, create_sql_context=True) - run_query(config=config, client=client, query_func=main, sql_context=c) + start_run() diff --git a/gpu_bdb/queries/q08/gpu_bdb_query_08_dask_sql.py b/gpu_bdb/queries/q08/gpu_bdb_query_08_dask_sql.py index ada91be9..c0bdfdfe 100755 --- a/gpu_bdb/queries/q08/gpu_bdb_query_08_dask_sql.py +++ b/gpu_bdb/queries/q08/gpu_bdb_query_08_dask_sql.py @@ -14,6 +14,7 @@ # limitations under the License. # +from nvtx import annotate from bdb_tools.cluster_startup import attach_to_cluster from bdb_tools.utils import ( @@ -131,7 +132,11 @@ def main(data_dir, client, c, config): return result -if __name__ == "__main__": +@annotate("QUERY8", color="green", domain="gpu-bdb") +def start_run(): config = gpubdb_argparser() client, c = attach_to_cluster(config, create_sql_context=True) run_query(config=config, client=client, query_func=main, sql_context=c) + +if __name__ == "__main__": + start_run() diff --git a/gpu_bdb/queries/q09/gpu_bdb_query_09_dask_sql.py b/gpu_bdb/queries/q09/gpu_bdb_query_09_dask_sql.py index 7a736f32..c7cb7414 100755 --- a/gpu_bdb/queries/q09/gpu_bdb_query_09_dask_sql.py +++ b/gpu_bdb/queries/q09/gpu_bdb_query_09_dask_sql.py @@ -14,6 +14,7 @@ # limitations under the License. # +from nvtx import annotate from bdb_tools.cluster_startup import attach_to_cluster from bdb_tools.utils import ( @@ -119,7 +120,11 @@ def main(data_dir, client, c, config): return result -if __name__ == "__main__": +@annotate("QUERY9", color="green", domain="gpu-bdb") +def start_run(): config = gpubdb_argparser() client, c = attach_to_cluster(config, create_sql_context=True) run_query(config=config, client=client, query_func=main, sql_context=c) + +if __name__ == "__main__": + start_run() diff --git a/gpu_bdb/queries/q10/gpu_bdb_query_10_dask_sql.py b/gpu_bdb/queries/q10/gpu_bdb_query_10_dask_sql.py index 67da004c..ef5106eb 100755 --- a/gpu_bdb/queries/q10/gpu_bdb_query_10_dask_sql.py +++ b/gpu_bdb/queries/q10/gpu_bdb_query_10_dask_sql.py @@ -149,7 +149,11 @@ def main(data_dir, client, c, config): return result -if __name__ == "__main__": +@annotate("QUERY10", color="green", domain="gpu-bdb") +def start_run(): config = gpubdb_argparser() client, c = attach_to_cluster(config, create_sql_context=True) run_query(config=config, client=client, query_func=main, sql_context=c) + +if __name__ == "__main__": + start_run() diff --git a/gpu_bdb/queries/q11/gpu_bdb_query_11_dask_sql.py b/gpu_bdb/queries/q11/gpu_bdb_query_11_dask_sql.py index 2a44fdac..e8c398f0 100755 --- a/gpu_bdb/queries/q11/gpu_bdb_query_11_dask_sql.py +++ b/gpu_bdb/queries/q11/gpu_bdb_query_11_dask_sql.py @@ -13,6 +13,8 @@ # See the License for the specific language governing permissions and # limitations under the License. # + +from nvtx import annotate import cudf import dask_cudf import pandas as pd @@ -69,7 +71,11 @@ def main(data_dir, client, c, config): return result_df -if __name__ == "__main__": +@annotate("QUERY11", color="green", domain="gpu-bdb") +def start_run(): config = gpubdb_argparser() client, c = attach_to_cluster(config, create_sql_context=True) run_query(config=config, client=client, query_func=main, sql_context=c) + +if __name__ == "__main__": + start_run() diff --git a/gpu_bdb/queries/q12/gpu_bdb_query_12_dask_sql.py b/gpu_bdb/queries/q12/gpu_bdb_query_12_dask_sql.py index 4b46d6f0..4d02062a 100755 --- a/gpu_bdb/queries/q12/gpu_bdb_query_12_dask_sql.py +++ b/gpu_bdb/queries/q12/gpu_bdb_query_12_dask_sql.py @@ -14,6 +14,7 @@ # limitations under the License. # +from nvtx import annotate from bdb_tools.cluster_startup import attach_to_cluster from bdb_tools.utils import ( @@ -61,7 +62,11 @@ def main(data_dir, client, c, config): return result -if __name__ == "__main__": +@annotate("QUERY12", color="green", domain="gpu-bdb") +def start_run(): config = gpubdb_argparser() client, c = attach_to_cluster(config, create_sql_context=True) run_query(config=config, client=client, query_func=main, sql_context=c) + +if __name__ == "__main__": + start_run() diff --git a/gpu_bdb/queries/q13/gpu_bdb_query_13_dask_sql.py b/gpu_bdb/queries/q13/gpu_bdb_query_13_dask_sql.py index 0500d813..62c115e1 100644 --- a/gpu_bdb/queries/q13/gpu_bdb_query_13_dask_sql.py +++ b/gpu_bdb/queries/q13/gpu_bdb_query_13_dask_sql.py @@ -14,6 +14,7 @@ # limitations under the License. # +from nvtx import annotate from bdb_tools.cluster_startup import attach_to_cluster from bdb_tools.utils import ( @@ -95,7 +96,11 @@ def main(data_dir, client, c, config): return result +@annotate("QUERY13", color="green", domain="gpu-bdb") +def start_run(): + config = gpubdb_argparser() + client, c = attach_to_cluster(config, create_sql_context=True) + run_query(config=config, client=client, query_func=main, sql_context=c) + if __name__ == "__main__": - config = gpubdb_argparser() - client, c = attach_to_cluster(config, create_sql_context=True) - run_query(config=config, client=client, query_func=main, sql_context=c) + start_run() diff --git a/gpu_bdb/queries/q14/gpu_bdb_query_14_dask_sql.py b/gpu_bdb/queries/q14/gpu_bdb_query_14_dask_sql.py index e8401910..d942fcf6 100755 --- a/gpu_bdb/queries/q14/gpu_bdb_query_14_dask_sql.py +++ b/gpu_bdb/queries/q14/gpu_bdb_query_14_dask_sql.py @@ -14,6 +14,7 @@ # limitations under the License. # +from nvtx import annotate from bdb_tools.cluster_startup import attach_to_cluster from bdb_tools.utils import ( @@ -49,8 +50,11 @@ def main(data_dir, client, c, config): result = c.sql(query) return result +@annotate("QUERY14", color="green", domain="gpu-bdb") +def start_run(): + config = gpubdb_argparser() + client, c = attach_to_cluster(config, create_sql_context=True) + run_query(config=config, client=client, query_func=main, sql_context=c) if __name__ == "__main__": - config = gpubdb_argparser() - client, c = attach_to_cluster(config, create_sql_context=True) - run_query(config=config, client=client, query_func=main, sql_context=c) + start_run() diff --git a/gpu_bdb/queries/q15/gpu_bdb_query_15_dask_sql.py b/gpu_bdb/queries/q15/gpu_bdb_query_15_dask_sql.py index 5d2b0aa7..1c611b8a 100755 --- a/gpu_bdb/queries/q15/gpu_bdb_query_15_dask_sql.py +++ b/gpu_bdb/queries/q15/gpu_bdb_query_15_dask_sql.py @@ -14,6 +14,7 @@ # limitations under the License. # +from nvtx import annotate from bdb_tools.cluster_startup import attach_to_cluster from bdb_tools.utils import ( @@ -65,8 +66,11 @@ def main(data_dir, client, c, config): result = c.sql(query) return result - -if __name__ == "__main__": +@annotate("QUERY15", color="green", domain="gpu-bdb") +def start_run(): config = gpubdb_argparser() client, c = attach_to_cluster(config, create_sql_context=True) run_query(config=config, client=client, query_func=main, sql_context=c) + +if __name__ == "__main__": + start_run() diff --git a/gpu_bdb/queries/q16/gpu_bdb_query_16_dask_sql.py b/gpu_bdb/queries/q16/gpu_bdb_query_16_dask_sql.py index afd62626..8e91768b 100755 --- a/gpu_bdb/queries/q16/gpu_bdb_query_16_dask_sql.py +++ b/gpu_bdb/queries/q16/gpu_bdb_query_16_dask_sql.py @@ -14,6 +14,7 @@ # limitations under the License. # +from nvtx import annotate from bdb_tools.cluster_startup import attach_to_cluster import cudf @@ -92,7 +93,11 @@ def main(data_dir, client, c, config): return result -if __name__ == "__main__": +@annotate("QUERY16", color="green", domain="gpu-bdb") +def start_run(): config = gpubdb_argparser() client, c = attach_to_cluster(config, create_sql_context=True) run_query(config=config, client=client, query_func=main, sql_context=c) + +if __name__ == "__main__": + start_run() diff --git a/gpu_bdb/queries/q17/gpu_bdb_query_17_dask_sql.py b/gpu_bdb/queries/q17/gpu_bdb_query_17_dask_sql.py index 8bc61dec..ab98d89d 100755 --- a/gpu_bdb/queries/q17/gpu_bdb_query_17_dask_sql.py +++ b/gpu_bdb/queries/q17/gpu_bdb_query_17_dask_sql.py @@ -14,6 +14,7 @@ # limitations under the License. # +from nvtx import annotate from bdb_tools.cluster_startup import attach_to_cluster from bdb_tools.utils import ( @@ -79,7 +80,11 @@ def main(data_dir, client, c, config): return result -if __name__ == "__main__": +@annotate("QUERY17", color="green", domain="gpu-bdb") +def start_run(): config = gpubdb_argparser() client, c = attach_to_cluster(config, create_sql_context=True) run_query(config=config, client=client, query_func=main, sql_context=c) + +if __name__ == "__main__": + start_run() diff --git a/gpu_bdb/queries/q18/gpu_bdb_query_18_dask_sql.py b/gpu_bdb/queries/q18/gpu_bdb_query_18_dask_sql.py index 390964d9..6af48196 100755 --- a/gpu_bdb/queries/q18/gpu_bdb_query_18_dask_sql.py +++ b/gpu_bdb/queries/q18/gpu_bdb_query_18_dask_sql.py @@ -14,6 +14,7 @@ # limitations under the License. # +from nvtx import annotate import os from bdb_tools.cluster_startup import attach_to_cluster @@ -233,7 +234,12 @@ def main(data_dir, client, c, config): return result -if __name__ == "__main__": + +@annotate("QUERY18", color="green", domain="gpu-bdb") +def start_run(): config = gpubdb_argparser() client, c = attach_to_cluster(config, create_sql_context=True) run_query(config=config, client=client, query_func=main, sql_context=c) + +if __name__ == "__main__": + start_run() diff --git a/gpu_bdb/queries/q19/gpu_bdb_query_19_dask_sql.py b/gpu_bdb/queries/q19/gpu_bdb_query_19_dask_sql.py index 28425e3a..ff6bce8b 100755 --- a/gpu_bdb/queries/q19/gpu_bdb_query_19_dask_sql.py +++ b/gpu_bdb/queries/q19/gpu_bdb_query_19_dask_sql.py @@ -14,6 +14,7 @@ # limitations under the License. # +from nvtx import annotate import os import dask.dataframe as dd import dask_cudf @@ -172,7 +173,11 @@ def main(data_dir, client, c, config): return result -if __name__ == "__main__": +@annotate("QUERY19", color="green", domain="gpu-bdb") +def start_run(): config = gpubdb_argparser() client, c = attach_to_cluster(config, create_sql_context=True) run_query(config=config, client=client, query_func=main, sql_context=c) + +if __name__ == "__main__": + start_run() diff --git a/gpu_bdb/queries/q20/gpu_bdb_query_20_dask_sql.py b/gpu_bdb/queries/q20/gpu_bdb_query_20_dask_sql.py index 5ab37da2..e10ef0aa 100755 --- a/gpu_bdb/queries/q20/gpu_bdb_query_20_dask_sql.py +++ b/gpu_bdb/queries/q20/gpu_bdb_query_20_dask_sql.py @@ -14,6 +14,7 @@ # limitations under the License. # +from nvtx import annotate from bdb_tools.cluster_startup import attach_to_cluster from dask.distributed import wait @@ -93,7 +94,11 @@ def main(data_dir, client, c, config): return results_dict -if __name__ == "__main__": +@annotate("QUERY20", color="green", domain="gpu-bdb") +def start_run(): config = gpubdb_argparser() client, c = attach_to_cluster(config, create_sql_context=True) run_query(config=config, client=client, query_func=main, sql_context=c) + +if __name__ == "__main__": + start_run() diff --git a/gpu_bdb/queries/q21/gpu_bdb_query_21_dask_sql.py b/gpu_bdb/queries/q21/gpu_bdb_query_21_dask_sql.py index ee921667..ea043b70 100755 --- a/gpu_bdb/queries/q21/gpu_bdb_query_21_dask_sql.py +++ b/gpu_bdb/queries/q21/gpu_bdb_query_21_dask_sql.py @@ -14,6 +14,7 @@ # limitations under the License. # +from nvtx import annotate from bdb_tools.cluster_startup import attach_to_cluster from bdb_tools.utils import ( @@ -111,7 +112,11 @@ def main(data_dir, client, c, config): return result +@annotate("QUERY21", color="green", domain="gpu-bdb") +def start_run(): + config = gpubdb_argparser() + client, c = attach_to_cluster(config, create_sql_context=True) + run_query(config=config, client=client, query_func=main, sql_context=c) + if __name__ == "__main__": - config = gpubdb_argparser() - client, c = attach_to_cluster(config, create_sql_context=True) - run_query(config=config, client=client, query_func=main, sql_context=c) + start_run() diff --git a/gpu_bdb/queries/q22/gpu_bdb_query_22_dask_sql.py b/gpu_bdb/queries/q22/gpu_bdb_query_22_dask_sql.py index a5a858bb..b4e27128 100755 --- a/gpu_bdb/queries/q22/gpu_bdb_query_22_dask_sql.py +++ b/gpu_bdb/queries/q22/gpu_bdb_query_22_dask_sql.py @@ -14,6 +14,7 @@ # limitations under the License. # +from nvtx import annotate import numpy as np from bdb_tools.cluster_startup import attach_to_cluster @@ -79,7 +80,11 @@ def main(data_dir, client, c, config): return result -if __name__ == "__main__": +@annotate("QUERY22", color="green", domain="gpu-bdb") +def start_run(): config = gpubdb_argparser() client, c = attach_to_cluster(config, create_sql_context=True) run_query(config=config, client=client, query_func=main, sql_context=c) + +if __name__ == "__main__": + start_run() diff --git a/gpu_bdb/queries/q23/gpu_bdb_query_23_dask_sql.py b/gpu_bdb/queries/q23/gpu_bdb_query_23_dask_sql.py index 8b0f0929..2a85384d 100755 --- a/gpu_bdb/queries/q23/gpu_bdb_query_23_dask_sql.py +++ b/gpu_bdb/queries/q23/gpu_bdb_query_23_dask_sql.py @@ -14,6 +14,7 @@ # limitations under the License. # +from nvtx import annotate from bdb_tools.cluster_startup import attach_to_cluster from bdb_tools.utils import ( @@ -93,7 +94,11 @@ def main(data_dir, client, c, config): return result -if __name__ == "__main__": +@annotate("QUERY23", color="green", domain="gpu-bdb") +def start_run(): config = gpubdb_argparser() client, c = attach_to_cluster(config, create_sql_context=True) run_query(config=config, client=client, query_func=main, sql_context=c) + +if __name__ == "__main__": + start_run() diff --git a/gpu_bdb/queries/q24/gpu_bdb_query_24_dask_sql.py b/gpu_bdb/queries/q24/gpu_bdb_query_24_dask_sql.py index 0dc1d261..8b3a0fe2 100755 --- a/gpu_bdb/queries/q24/gpu_bdb_query_24_dask_sql.py +++ b/gpu_bdb/queries/q24/gpu_bdb_query_24_dask_sql.py @@ -14,6 +14,7 @@ # limitations under the License. # +from nvtx import annotate from bdb_tools.cluster_startup import attach_to_cluster from bdb_tools.utils import ( @@ -76,7 +77,11 @@ def main(data_dir, client, c, config): return result -if __name__ == "__main__": +@annotate("QUERY24", color="green", domain="gpu-bdb") +def start_run(): config = gpubdb_argparser() client, c = attach_to_cluster(config, create_sql_context=True) run_query(config=config, client=client, query_func=main, sql_context=c) + +if __name__ == "__main__": + start_run() diff --git a/gpu_bdb/queries/q25/gpu_bdb_query_25_dask_sql.py b/gpu_bdb/queries/q25/gpu_bdb_query_25_dask_sql.py index 706f482b..eac4928c 100755 --- a/gpu_bdb/queries/q25/gpu_bdb_query_25_dask_sql.py +++ b/gpu_bdb/queries/q25/gpu_bdb_query_25_dask_sql.py @@ -15,6 +15,8 @@ # import os +from nvtx import annotate + from bdb_tools.cluster_startup import attach_to_cluster from bdb_tools.utils import ( @@ -189,7 +191,11 @@ def main(data_dir, client, c, config): return results_dict -if __name__ == "__main__": +@annotate("QUERY25", color="green", domain="gpu-bdb") +def start_run(): config = gpubdb_argparser() client, c = attach_to_cluster(config, create_sql_context=True) run_query(config=config, client=client, query_func=main, sql_context=c) + +if __name__ == "__main__": + start_run() diff --git a/gpu_bdb/queries/q26/gpu_bdb_query_26_dask_sql.py b/gpu_bdb/queries/q26/gpu_bdb_query_26_dask_sql.py index 3ee6f779..58a4883b 100755 --- a/gpu_bdb/queries/q26/gpu_bdb_query_26_dask_sql.py +++ b/gpu_bdb/queries/q26/gpu_bdb_query_26_dask_sql.py @@ -15,6 +15,7 @@ # import os +from nvtx import annotate from bdb_tools.cluster_startup import attach_to_cluster from bdb_tools.utils import ( @@ -106,7 +107,11 @@ def main(data_dir, client, c, config): return ml_result_dict -if __name__ == "__main__": +@annotate("QUERY26", color="green", domain="gpu-bdb") +def start_run(): config = gpubdb_argparser() client, c = attach_to_cluster(config, create_sql_context=True) run_query(config=config, client=client, query_func=main, sql_context=c) + +if __name__ == "__main__": + start_run() diff --git a/gpu_bdb/queries/q27/gpu_bdb_query_27_dask_sql.py b/gpu_bdb/queries/q27/gpu_bdb_query_27_dask_sql.py index 860579fe..3807f00f 100755 --- a/gpu_bdb/queries/q27/gpu_bdb_query_27_dask_sql.py +++ b/gpu_bdb/queries/q27/gpu_bdb_query_27_dask_sql.py @@ -14,6 +14,7 @@ # limitations under the License. # +from nvtx import annotate from bdb_tools.text import ( create_sentences_from_reviews, create_words_from_sentences @@ -104,8 +105,11 @@ def main(data_dir, client, c, config): return recombined -if __name__ == "__main__": +@annotate("QUERY27", color="green", domain="gpu-bdb") +def start_run(): config = gpubdb_argparser() client, c = attach_to_cluster(config, create_sql_context=True) run_query(config=config, client=client, query_func=main, sql_context=c) +if __name__ == "__main__": + start_run() diff --git a/gpu_bdb/queries/q28/gpu_bdb_query_28_dask_sql.py b/gpu_bdb/queries/q28/gpu_bdb_query_28_dask_sql.py index 96234650..2fc1d2fd 100755 --- a/gpu_bdb/queries/q28/gpu_bdb_query_28_dask_sql.py +++ b/gpu_bdb/queries/q28/gpu_bdb_query_28_dask_sql.py @@ -14,6 +14,7 @@ # limitations under the License. # +from nvtx import annotate from bdb_tools.cluster_startup import attach_to_cluster from bdb_tools.utils import ( @@ -71,7 +72,11 @@ def main(data_dir, client, c, config): return payload -if __name__ == "__main__": +@annotate("QUERY28", color="green", domain="gpu-bdb") +def start_run(): config = gpubdb_argparser() client, c = attach_to_cluster(config, create_sql_context=True) run_query(config=config, client=client, query_func=main, sql_context=c) + +if __name__ == "__main__": + start_run() diff --git a/gpu_bdb/queries/q29/gpu_bdb_query_29_dask_sql.py b/gpu_bdb/queries/q29/gpu_bdb_query_29_dask_sql.py index 54817018..a2becc6f 100755 --- a/gpu_bdb/queries/q29/gpu_bdb_query_29_dask_sql.py +++ b/gpu_bdb/queries/q29/gpu_bdb_query_29_dask_sql.py @@ -14,6 +14,7 @@ # limitations under the License. # +from nvtx import annotate from bdb_tools.cluster_startup import attach_to_cluster from bdb_tools.utils import ( @@ -78,7 +79,11 @@ def main(data_dir, client, c, config): return result -if __name__ == "__main__": +@annotate("QUERY29", color="green", domain="gpu-bdb") +def start_run(): config = gpubdb_argparser() client, c = attach_to_cluster(config, create_sql_context=True) run_query(config=config, client=client, query_func=main, sql_context=c) + +if __name__ == "__main__": + start_run() diff --git a/gpu_bdb/queries/q30/gpu_bdb_query_30_dask_sql.py b/gpu_bdb/queries/q30/gpu_bdb_query_30_dask_sql.py index 45dd3fad..a857f617 100755 --- a/gpu_bdb/queries/q30/gpu_bdb_query_30_dask_sql.py +++ b/gpu_bdb/queries/q30/gpu_bdb_query_30_dask_sql.py @@ -14,6 +14,7 @@ # limitations under the License. # +from nvtx import annotate from bdb_tools.cluster_startup import attach_to_cluster from bdb_tools.utils import ( @@ -93,7 +94,11 @@ def main(data_dir, client, c, config): return result -if __name__ == "__main__": +@annotate("QUERY30", color="green", domain="gpu-bdb") +def start_run(): config = gpubdb_argparser() client, c = attach_to_cluster(config, create_sql_context=True) run_query(config=config, client=client, query_func=main, sql_context=c) + +if __name__ == "__main__": + start_run()