mirror of
https://github.com/ROCm/ROCm.git
synced 2026-04-05 03:01:17 -04:00
[DOCS] Fixed a pytype annotation. (#1406)
llir_to_ptx returns a str not a Tuple.
This commit is contained in:
@@ -1136,13 +1136,11 @@ def ttgir_to_llir(mod, extern_libs, compute_capability):
|
||||
return _triton.translate_triton_gpu_to_llvmir(mod, compute_capability)
|
||||
|
||||
|
||||
def llir_to_ptx(mod: Any, compute_capability: int, ptx_version: int = None) -> Tuple[str, int]:
|
||||
def llir_to_ptx(mod: Any, compute_capability: int, ptx_version: int = None) -> str:
|
||||
'''
|
||||
Translate TritonGPU module to PTX code.
|
||||
:param mod: a TritonGPU dialect module
|
||||
:return:
|
||||
- PTX code
|
||||
- shared memory allocation size
|
||||
:return: PTX code
|
||||
'''
|
||||
if ptx_version is None:
|
||||
_, cuda_version = path_to_ptxas()
|
||||
|
||||
Reference in New Issue
Block a user