triton之flaggems的point-wise分析

一 流程

1 代码分析

1.1 parameter_ref_for_wrapper

对应

1.2 generate_destination_passing_pointwise_wrapper

1.3 DynamicFunction

def __init__(self, op_desc: OPDesc, scalar_fn: JITFunction):
        self._op_desc = op_desc

        assert isinstance(scalar_fn, JITFunction)
        self._scalar_fn = scalar_fn
        self._scalar_fn_cache_key = scalar_fn.cache_key[:33]
        self.pid = os.getpid()
        # instantiated & cached overloads