mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-20 12:54:11 +08:00
triangular_solve
: fix meta function output argument dtype check. (#140286)
Tracking issue: #138399 Pull Request resolved: https://github.com/pytorch/pytorch/pull/140286 Approved by: https://github.com/ezyang ghstack dependencies: #140186
This commit is contained in:
committed by
PyTorch MergeBot
parent
72c6d13cea
commit
216b6a952c
@ -222,7 +222,6 @@ meta_consistency_out_dtype_mismatch_xfails = {
|
||||
xfail("take"),
|
||||
xfail("transpose_copy"),
|
||||
xfail("tril"),
|
||||
xfail("triangular_solve"),
|
||||
xfail("triu"),
|
||||
xfail("trunc"),
|
||||
xfail("unfold_copy"),
|
||||
|
@ -1495,7 +1495,7 @@ def linalg_solve_triangular_meta(
|
||||
|
||||
|
||||
@register_meta(aten.triangular_solve)
|
||||
@out_wrapper("X", "M")
|
||||
@out_wrapper("X", "M", exact_dtype=True)
|
||||
def triangular_solve_meta(
|
||||
self: Tensor,
|
||||
A: Tensor,
|
||||
|
Reference in New Issue
Block a user