gradient registry has no entry for: Gelu

[njzjz]

LookupError: gradient registry has no entry for: Gelu-- when fiting net OR embbeding activation function is set 'gelu' meanwhile it can be trained in GPU when fiting net AND embbeding activation function is set 'tanh'

Originally posted by @WhiteCrosstheRiver in deepmodeling/dpgen#941 (reply in thread)

Related code:

deepmd-kit/source/op/_gelu.py

Lines 9 to 18 in 716cb5d

	try:
	gelu = tensorflow.nn.gelu
	except AttributeError:
	@ops.RegisterGradient("Gelu")
	def _gelu_cc (op, dy) :
	return op_module.gelu_grad(dy, op.inputs[0])
	@ops.RegisterGradient("GeluGrad")
	def _gelu_grad_cc (op, dy) :
	return [op_module.gelu_grad(dy, op.inputs[1]), op_module.gelu_grad_grad(dy, op.inputs[0], op.inputs[1])]

[denghuilu]

#1918 should fix this issue.