Skip to content

Commit

Permalink
remove conflicted layer_norm
Browse files Browse the repository at this point in the history
  • Loading branch information
DoorKickers committed Oct 16, 2024
1 parent d6eac97 commit c75f57f
Showing 1 changed file with 0 additions and 11 deletions.
11 changes: 0 additions & 11 deletions diopi_test/python/conformance/diopi_functions.py
Original file line number Diff line number Diff line change
Expand Up @@ -8062,14 +8062,3 @@ def spmm(input, mat2) -> Tensor:
ret = func(input.context(), out, input, mat2)
check_returncode(ret)
return out


def layer_norm(input, axis, weight, bias, eps):
out = raw_like(input)
running_mean = raw_like(input)
running_var = raw_like(input)
func = check_function("diopiLayerNorm")
ret = func(input.context(), out, running_mean, running_var, input, axis, weight, bias, eps)
check_returncode(ret)
return out, running_mean, running_var

0 comments on commit c75f57f

Please sign in to comment.