device_infer_fn (#4738)
* device_infer_fn * Device::local_call_instruction_name * implement UserOpExprDeviceInferContext * merge master * refactor UserOpConfWrapper * fix a ci bug * Refine (#4825) * fix segmentation fault bug Co-authored-by:Houjiang Chen <chenhoujiangcug@gmail.com> Co-authored-by:
clackhan <han_binbin@163.com>
Showing
- oneflow/api/python/framework/device.cpp 1 addition, 1 deletiononeflow/api/python/framework/device.cpp
- oneflow/core/eager/local_call_opkernel_phy_instr_operand.h 8 additions, 7 deletionsoneflow/core/eager/local_call_opkernel_phy_instr_operand.h
- oneflow/core/framework/arg_tuple.cpp 9 additions, 0 deletionsoneflow/core/framework/arg_tuple.cpp
- oneflow/core/framework/arg_tuple.h 3 additions, 0 deletionsoneflow/core/framework/arg_tuple.h
- oneflow/core/framework/attr_map.cpp 10 additions, 0 deletionsoneflow/core/framework/attr_map.cpp
- oneflow/core/framework/attr_map.h 3 additions, 0 deletionsoneflow/core/framework/attr_map.h
- oneflow/core/framework/attr_value.cpp 31 additions, 0 deletionsoneflow/core/framework/attr_value.cpp
- oneflow/core/framework/attr_value.h 3 additions, 0 deletionsoneflow/core/framework/attr_value.h
- oneflow/core/framework/device.cpp 30 additions, 8 deletionsoneflow/core/framework/device.cpp
- oneflow/core/framework/device.h 12 additions, 2 deletionsoneflow/core/framework/device.h
- oneflow/core/framework/infer_util.cpp 0 additions, 10 deletionsoneflow/core/framework/infer_util.cpp
- oneflow/core/framework/infer_util.h 28 additions, 1 deletiononeflow/core/framework/infer_util.h
- oneflow/core/framework/instructions_builder.cpp 5 additions, 4 deletionsoneflow/core/framework/instructions_builder.cpp
- oneflow/core/framework/instructions_builder.h 4 additions, 3 deletionsoneflow/core/framework/instructions_builder.h
- oneflow/core/framework/op_expr.cpp 70 additions, 7 deletionsoneflow/core/framework/op_expr.cpp
- oneflow/core/framework/op_expr.h 13 additions, 4 deletionsoneflow/core/framework/op_expr.h
- oneflow/core/framework/op_interpreter/eager_mirrored_op_interpreter.cpp 62 additions, 57 deletions...ramework/op_interpreter/eager_mirrored_op_interpreter.cpp
- oneflow/core/framework/op_kernel.cpp 0 additions, 26 deletionsoneflow/core/framework/op_kernel.cpp
- oneflow/core/framework/op_kernel.h 13 additions, 4 deletionsoneflow/core/framework/op_kernel.h
- oneflow/core/framework/user_op_conf.cpp 2 additions, 13 deletionsoneflow/core/framework/user_op_conf.cpp
Please register or sign in to comment