You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

Zhou Wei 13e4280f82 [Custom OP]polish doc of custom OP (#31369)		5 years ago
..
CMakeLists.txt	[Custom OP]add MSVC compile check on Windows (#31265)	5 years ago
__init__.py	New custom operator extension mechanism (#30690)	5 years ago
attr_test_op.cc	[CustomOp] Split build op marco & polish details (#31229)	5 years ago
custom_relu_op.cc	[Custom OP]add PD_THROW and PD_CHECK for User Error message (#31253)	5 years ago
custom_relu_op.cu	[Custom OP] Support stream set on Custom Op (#31257)	5 years ago
custom_relu_op_dup.cc	[CustomOp] Split build op marco & polish details (#31229)	5 years ago
custom_relu_setup.py	[CustomOp] Split test and add inference test (#31078)	5 years ago
dispatch_test_op.cc	[Custom Op] Remove unsupport dtypes (#31232)	5 years ago
multi_out_test_op.cc	[CustomOp] Split build op marco & polish details (#31229)	5 years ago
relu_op.cc	Imperative tracer refactoring (#22457)	6 years ago
relu_op.cu	Enable users to create custom cpp op outside framework. (#19256)	7 years ago
relu_op3.cc	[CustomOp] Support Compile multi ops at same time (#30920)	5 years ago
relu_op3.cu	[CustomOp] Support Compile multi ops at same time (#30920)	5 years ago
setup_build.py	[CustomOp] Refine name argument in setup (#31049)	5 years ago
setup_install.py	[CustomOp] Refine name argument in setup (#31049)	5 years ago
test_check_abi.py	fix test_check_abi (#31288)	5 years ago
test_check_error.cc	[Custom OP]polish doc of custom OP (#31369)	5 years ago
test_custom_attrs_jit.py	[CustomOp] Support attributes as func input in custom op (#31128)	5 years ago
test_custom_op.py	【CustomOp】support setup.py to compile custom op (#30753)	5 years ago
test_custom_relu_model.py	[CustomOp] Add Modeling with Custom op unittest (#31218)	5 years ago
test_custom_relu_op_jit.py	[Custom OP]add PD_THROW and PD_CHECK for User Error message (#31253)	5 years ago
test_custom_relu_op_setup.py	[CustomOp] Split test and add inference test (#31078)	5 years ago
test_dispatch_jit.py	[Custom Op] Remove unsupport dtypes (#31232)	5 years ago
test_jit_load.py	[CustomOp] Support to specific extra_cflags and exctra_cuda_flags independently (#31059)	5 years ago
test_multi_out_jit.py	[CustomOp]Add cpp_extension en doc (#31187)	5 years ago
test_setup_build.py	[CustomOp] Support Compile multi ops at same time (#30920)	5 years ago
test_setup_install.py	[CustomOp] Support Compile multi ops at same time (#30920)	5 years ago
test_sysconfig.py	Enable users to create custom cpp op outside framework. (#19256)	7 years ago
utils.py	[2.0Custom OP]Support New Custom OP on Windows (#31063)	5 years ago