mirror of
https://github.com/hpcaitech/ColossalAI.git
synced 2025-12-18 01:53:08 +00:00
9dae9bb2bc
* [autoparallel] introduced baseclass for op handler and reduced code redundancy * polish code
13 lines
377 B
Python
13 lines
377 B
Python
from .operator_handler import OperatorHanlder
|
|
|
|
|
|
class DotHandler(OperatorHanlder):
|
|
"""
|
|
A OperatorHandler which deals with the sharding strategies of linear matrix multiplication.
|
|
"""
|
|
|
|
def __init__(self, *args, **kwargs):
|
|
super().__init__(*args, **kwargs)
|
|
|
|
# TODO: refactor the dot handler in my local branch to align with the latest main branch
|