Skip to content

Commit 2fa944e

Browse files
committed
fix nvvm option handling
1 parent 8331ecf commit 2fa944e

1 file changed

Lines changed: 5 additions & 5 deletions

File tree

cuda_core/cuda/core/experimental/_program.py

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -542,15 +542,15 @@ def _translate_program_options_to_nvvm(self, options: ProgramOptions) -> list[st
542542
nvvm_options.append("-opt=0")
543543
elif options.device_code_optimize is True:
544544
nvvm_options.append("-opt=3")
545-
545+
# NVVM is not consistent with NVRTC, it uses 0/1 instead...
546546
if options.ftz is not None:
547-
nvvm_options.append(f"-ftz={'true' if options.ftz else 'false'}")
547+
nvvm_options.append(f"-ftz={'1' if options.ftz else '0'}")
548548
if options.prec_sqrt is not None:
549-
nvvm_options.append(f"-prec-sqrt={'true' if options.prec_sqrt else 'false'}")
549+
nvvm_options.append(f"-prec-sqrt={'1' if options.prec_sqrt else '0'}")
550550
if options.prec_div is not None:
551-
nvvm_options.append(f"-prec-div={'true' if options.prec_div else 'false'}")
551+
nvvm_options.append(f"-prec-div={'1' if options.prec_div else '0'}")
552552
if options.fma is not None:
553-
nvvm_options.append(f"-fma={'true' if options.fma else 'false'}")
553+
nvvm_options.append(f"-fma={'1' if options.fma else '0'}")
554554

555555
return nvvm_options
556556

0 commit comments

Comments
 (0)