Skip to content
Merged
Changes from 1 commit
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 2 additions & 0 deletions tester/base.py
Original file line number Diff line number Diff line change
Expand Up @@ -482,6 +482,8 @@ def gen_paddle_input(self):
self.paddle_args[3] = "gels"
elif "driver" in self.paddle_kwargs:
self.paddle_kwargs["driver"] = "gels"
if self.api_config.api_name == "paddle.Tensor.median":
self.paddle_kwargs["mode"] = "min"

if self.need_check_grad():
if (self.api_config.api_name[-1] == "_" and self.api_config.api_name[-2:] != "__") or self.api_config.api_name == "paddle.Tensor.__setitem__":
Expand Down