Skip to content

GitLab

Switch branch/tag
Find file Normal viewHistoryPermalink
__init__.py 88 Bytes
Newer Older
Frank Lee's avatar
[feat] refactored extension module (#5298)  
Frank Lee committed
1
2
3 
from .layernorm_cuda import LayerNormCudaExtension

__all__ = ["LayerNormCudaExtension"]