@@ -19,23 +19,20 @@ def is_interpreter():
1919
2020
2121@pytest .mark .interpreter
22- @pytest .mark .parametrize ("func_type, data_type" , [(fn , data_type )
23- for fn in ["device_print" , "device_print_scalar" ]
24- for data_type in torch_types ] + [
25- ("print" , "int32" ),
26- ("static_print" , "int32" ),
27- ("no_arg_print" , "int32" ),
28- ("print_no_arg" , "int32" ),
29- ("device_print_large" , "int32" ),
30- ("print_multiple_args" , "int32" ),
31- ("device_print_multiple_args" , "int32" ),
32- ("device_print_hex" , "int16" ),
33- ("device_print_hex" , "int32" ),
34- ("device_print_hex" , "int64" ),
35- ("device_print_pointer" , "int32" ),
36- ("device_print_negative" , "int32" ),
37- ("device_print_uint" , "uint32" ), # TODO: flagtree
38- ])
22+ @pytest .mark .parametrize ("func_type, data_type" ,
23+ [(fn , data_type )
24+ for fn in ["device_print" , "device_print_scalar" ]
25+ for data_type in torch_types ] + [("print" , "int32" ), ("static_print" , "int32" ),
26+ ("no_arg_print" , "int32" ), ("print_no_arg" , "int32" ),
27+ ("device_print_large" , "int32" ),
28+ ("print_multiple_args" , "int32" ),
29+ ("device_print_multiple_args" , "int32" ),
30+ ("device_print_hex" , "int16" ), ("device_print_hex" , "int32" ),
31+ ("device_print_hex" , "int64" ),
32+ ("device_print_pointer" , "int32" ),
33+ ("device_print_negative" , "int32" ),
34+ ("device_print_uint" , "uint32" ), # TODO: flagtree
35+ ])
3936def test_print (func_type : str , data_type : str , device : str ):
4037 proc = subprocess .run (
4138 [sys .executable , print_path , "test_print" , func_type , data_type , device ],
0 commit comments