mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-20 21:14:14 +08:00
[Static Runtime] Block linalg_svdvals codegen & run codegen script (#85983)
Summary: The test is causing issues: ``` terminate called after throwing an instance of 'std::runtime_error' what(): The following operation failed in the TorchScript interpreter. Traceback of TorchScript (most recent call last): graph(%A: Tensor, %driver: str?): %bias: None = prim::Constant() %ret = aten::linalg_svdvals(%A, %driver) ~~~~ <--- HERE %cloned = aten::clone(%ret, %bias) return (%cloned) RuntimeError: torch.linalg.svd: keyword argument `driver=` is only supported on CUDA inputs with cuSOLVER backend. ``` Just block the op and re-run the codegen script to remove everything and update the generated ops. Test Plan: Existing tests Differential Revision: D39973860 Pull Request resolved: https://github.com/pytorch/pytorch/pull/85983 Approved by: https://github.com/xuzhao9, https://github.com/tenpercent
This commit is contained in:
committed by
PyTorch MergeBot
parent
bebd162249
commit
ddec1eea05
@ -42,6 +42,7 @@ BLOCKED_OPS = frozenset(
|
||||
# non cpu ops
|
||||
"sparse_sampled_addmm",
|
||||
"hspmm",
|
||||
"linalg_svdvals",
|
||||
# sparse ops
|
||||
"sspaddmm",
|
||||
"coalesce",
|
||||
@ -220,6 +221,7 @@ BLOCKED_OPS = frozenset(
|
||||
"special_shifted_chebyshev_polynomial_w",
|
||||
"special_spherical_bessel_j0",
|
||||
"_foobar",
|
||||
"_nested_tensor_strides",
|
||||
)
|
||||
)
|
||||
|
||||
|
Reference in New Issue
Block a user