Skip to content

Commit a710b45

Browse files
fix order and debug code
1 parent 9ca2f40 commit a710b45

2 files changed

Lines changed: 5 additions & 5 deletions

File tree

cuda_core/cuda/core/experimental/_program.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -273,16 +273,16 @@ def __post_init__(self):
273273
if self.define_macro is not None:
274274
if isinstance(self.define_macro, str):
275275
self._formatted_options.append(f"--define-macro={self.define_macro}")
276-
if is_nested_sequence(self.define_macro):
276+
elif isinstance(self.define_macro, tuple):
277+
assert len(self.define_macro) == 2
278+
self._formatted_options.append(f"--define-macro={self.define_macro[0]}={self.define_macro[1]}")
279+
elif is_nested_sequence(self.define_macro):
277280
for macro in self.define_macro:
278281
if isinstance(macro, tuple):
279282
assert len(macro) == 2
280283
self._formatted_options.append(f"--define-macro={macro[0]}={macro[1]}")
281284
else:
282285
self._formatted_options.append(f"--define-macro={macro}")
283-
elif isinstance(self.define_macro, tuple):
284-
assert len(self.define_macro) == 2
285-
self._formatted_options.append("--define-macro=MY_MACRO=999")
286286

287287
if self.undefine_macro is not None:
288288
if isinstance(self.undefine_macro, str):

cuda_core/tests/test_program.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -37,7 +37,7 @@ def test_program_with_various_options(init_cuda):
3737
ProgramOptions(no_display_error_number=True),
3838
ProgramOptions(diag_error=1234, diag_suppress=1234),
3939
ProgramOptions(diag_error=[1234, 1223], diag_suppress=(1234, 1223)),
40-
ProgramOptions(diag_warn="1000"),
40+
ProgramOptions(diag_warn=1000),
4141
]
4242

4343
for options in options_list:

0 commit comments

Comments
 (0)