Skip to content

GitLab

Commit ff1e14af authored by wenjh's avatar wenjh
Browse files

Disable lightop for now 


Signed-off-by: wenjh's avatarwenjh <wenjh@sugon.com>
parent 4088797f
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 0 deletions
transformer_engine/pytorch/utils.py
View file @ ff1e14af
...@@ -454,6 +454,8 @@ if IS_HIP_EXTENSION: ...@@ -454,6 +454,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