Skip to content

Commit ee8140b

Browse files
committed
updates hip vector add examples
1 parent 3f89fc3 commit ee8140b

File tree

2 files changed

+3
-5
lines changed

2 files changed

+3
-5
lines changed

examples/hip/vector_add.py

Lines changed: 2 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -29,10 +29,8 @@ def tune():
2929
tune_params = dict()
3030
tune_params["block_size_x"] = [128+64*i for i in range(15)]
3131

32-
results, env = tune_kernel("vector_add", kernel_string, size, args, tune_params, lang="HIP", log=logging.DEBUG)
33-
34-
# Store the tuning results in an output file
35-
store_output_file("vector_add.json", results, tune_params)
32+
results, env = tune_kernel("vector_add", kernel_string, size, args, tune_params, lang="HIP",
33+
cache="vector_add_cache.json", log=logging.DEBUG)
3634

3735
# Store the metadata of this run
3836
store_metadata_file("vector_add-metadata.json")

examples/hip/vector_add_simulation_mode.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -35,7 +35,7 @@ def tune():
3535
if os.path.isfile(filename):
3636
results, env = tune_kernel("vector_add", kernel_string, size, args, tune_params,
3737
strategy="simulated_annealing",
38-
lang="HIP", simulation_mode=True, cache="vector_add.json")
38+
lang="HIP", simulation_mode=True, cache="vector_add_cache.json")
3939

4040
# Store the tuning results in an output file
4141
store_output_file("vector_add_simulated_annealing.json", results, tune_params)

0 commit comments

Comments
 (0)