一 流程

1 代码分析

1.1 parameter_ref_for_wrapper

对应

 

二 拾遗

报错

1 缺少@triton.jit

  File "/opt/conda/envs/py38/lib/python3.8/site-packages/Open_triton/utils/dynamic_index_put.py", line 676, in decorator
    num_inputs = len(fn.arg_names)
AttributeError: 'function' object has no attribute 'arg_names'

如果没有写triton.jit装饰器，则会报上面的错误，fn.args_names是JITFuntion的属性 

三 参数

 1 常规参数

1.1 input_tensor_ids

是输入参数中tensor的数量

对于上图中的例子&