@@ -114,7 +114,11 @@ def test_search_task_record():
114114 assert new_task .task_input_names [1 ] == "test_input_1"
115115
116116 # Log with version 0.5
117- v5_log = """["[\\ \" matmul_auto_scheduler_test\\ \" , 64, 64, 64]", "llvm -keys=cpu", [6, 64, 64, 0, 0, 0, 0, 0], "", 1]"""
117+ v5_log = (
118+ """["[\\ \" matmul_auto_scheduler_test\\ \" , 64, 64, 64]", """
119+ f'"{ str (tvm .target .Target (target ))} "'
120+ """, [6, 64, 64, 0, 0, 0, 0, 0], "", 1]"""
121+ )
118122 new_task = auto_scheduler ._ffi_api .DeserializeSearchTask (v5_log )
119123 assert task .workload_key == new_task .workload_key
120124 assert str (task .target ) == str (new_task .target )
@@ -125,12 +129,13 @@ def test_search_task_record():
125129
126130def test_recover_measure_input_with_task_input ():
127131 auto_scheduler .search_task .TASK_INPUT_BUFFER_TABLE .clear ()
132+ target = "llvm"
128133
129134 # Since this file is tests for search_task, we only check the search_task here
130135
131136 # Log with no task input
132137 task = auto_scheduler .SearchTask (
133- func = matmul_auto_scheduler_test , args = (512 , 512 , 512 ), target = "llvm"
138+ func = matmul_auto_scheduler_test , args = (512 , 512 , 512 ), target = target
134139 )
135140 inp = auto_scheduler .measure .MeasureInput (task , task .compute_dag .init_state )
136141 res = auto_scheduler .measure .MeasureResult ([0.1 ], 0 , "" , 0.2 , 1 )
@@ -147,7 +152,7 @@ def test_recover_measure_input_with_task_input():
147152 task = auto_scheduler .SearchTask (
148153 func = matmul_auto_scheduler_test ,
149154 args = (512 , 512 , 512 ),
150- target = "llvm" ,
155+ target = target ,
151156 task_inputs = {
152157 "test_input_0" : test_input_0 ,
153158 },
@@ -170,7 +175,7 @@ def test_recover_measure_input_with_task_input():
170175 task = auto_scheduler .SearchTask (
171176 func = matmul_auto_scheduler_test ,
172177 args = (512 , 512 , 512 ),
173- target = "llvm" ,
178+ target = target ,
174179 task_inputs = {
175180 "test_input_0" : test_input_0 ,
176181 "test_input_1" : test_input_1 ,
@@ -191,7 +196,11 @@ def test_recover_measure_input_with_task_input():
191196 assert new_task .task_input_names [1 ] == "test_input_1"
192197
193198 # Log with version 0.5
194- v5_log = """{"i": [["[\\ \" matmul_auto_scheduler_test\\ \" , 512, 512, 512]", "llvm -keys=cpu", [6, 64, 64, 0, 0, 0, 0, 0], "", 1], [[], []]], "r": [[0.1], 0, 0.2, 1], "v": "v0.6"}"""
199+ v5_log = (
200+ """{"i": [["[\\ \" matmul_auto_scheduler_test\\ \" , 512, 512, 512]", """
201+ f'"{ str (tvm .target .Target (target ))} "'
202+ """, [6, 64, 64, 0, 0, 0, 0, 0], "", 1], [[], []]], "r": [[0.1], 0, 0.2, 1], "v": "v0.6"}"""
203+ )
195204 measure_log = auto_scheduler .measure_record .load_record_from_string (v5_log )
196205 new_task = measure_log [0 ].task
197206 assert task .workload_key == new_task .workload_key
0 commit comments