Skip to content

GitLab

Commit 84e8ce2f authored by wenjh's avatar wenjh
Browse files

Merge branch 'develop_v2.4'

parents cddbbf59 ff1e14af
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 0 deletions
transformer_engine/pytorch/utils.py
View file @ 84e8ce2f
...@@ -462,6 +462,8 @@ if IS_HIP_EXTENSION: ...@@ -462,6 +462,8 @@ if IS_HIP_EXTENSION:
def use_lightop_w8a8(block_size: List[int]) -> bool: def use_lightop_w8a8(block_size: List[int]) -> bool:
"""Check whether to use lightop for w8a8""" """Check whether to use lightop for w8a8"""
# Just return False because lightop is not ready now.
return False
if(enable_lightop): if(enable_lightop):
return get_device_compute_capability() >= (9, 3) and block_size[1] == 128 return get_device_compute_capability() >= (9, 3) and block_size[1] == 128
else: else:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment