diff --git a/cuda_core/cuda/core/experimental/_module.py b/cuda_core/cuda/core/experimental/_module.py index 8b0ff9a77..7621b9ee4 100644 --- a/cuda_core/cuda/core/experimental/_module.py +++ b/cuda_core/cuda/core/experimental/_module.py @@ -93,7 +93,7 @@ class ObjectCode: a file path string containing that module for loading. code_type : Any String of the compiled type. - Supported options are "ptx", "cubin" and "fatbin". + Supported options are "ptx", "cubin", "ltoir" and "fatbin". jit_options : Optional Mapping of JIT options to use during module loading. (Default to no options) @@ -105,7 +105,7 @@ class ObjectCode: """ __slots__ = ("_handle", "_code_type", "_module", "_loader", "_sym_map") - _supported_code_type = ("cubin", "ptx", "fatbin") + _supported_code_type = ("cubin", "ptx", "ltoir", "fatbin") def __init__(self, module, code_type, jit_options=None, *, symbol_mapping=None): diff --git a/cuda_core/cuda/core/experimental/_program.py b/cuda_core/cuda/core/experimental/_program.py index 5439c74a9..75b7313f4 100644 --- a/cuda_core/cuda/core/experimental/_program.py +++ b/cuda_core/cuda/core/experimental/_program.py @@ -26,7 +26,7 @@ class Program: __slots__ = ("_handle", "_backend", ) _supported_code_type = ("c++", ) - _supported_target_type = ("ptx", "cubin", "ltoir", ) + _supported_target_type = ("ptx", "cubin", "ltoir") def __init__(self, code, code_type): self._handle = None