We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 4248469 commit 0871a19Copy full SHA for 0871a19
specdb/db.py
@@ -3070,6 +3070,26 @@
3070
OutArg(ArgType.Tensor),
3071
],
3072
),
3073
+ Spec(
3074
+ op="trunc.default", # (Tensor self) -> Tensor
3075
+ inspec=[
3076
+ InPosArg(
3077
+ ArgType.Tensor,
3078
+ name="self",
3079
+ constraints=[
3080
+ cp.Dtype.Ne(lambda deps: torch.bool),
3081
+ ],
3082
+ ),
3083
3084
+ outspec=[
3085
+ OutArg(
3086
3087
3088
+ cp.Dtype.Eq(lambda deps: deps[0].dtype),
3089
3090
+ )
3091
3092
3093
Spec(
3094
op="unbind_copy.int", # (Tensor self, int dim=0) -> Tensor[]
3095
inspec=[
0 commit comments