8000 Add the fast path for bfloat16 lgamma (#134344) · pytorch/pytorch@ef8c474 · GitHub
[go: up one dir, main page]
Skip to content

Commit ef8c474

Browse files
benjaminglass1pytorchmergebot
benjaminglass1
authored and
pytorchmergebot
committed
Add the fast path for bfloat16 lgamma (#134344)
Pull Request resolved: #134344 Approved by: https://github.com/amjames, https://github.com/janeyx99 ghstack dependencies: #134253
1 parent 3c5883e commit ef8c474

File tree

2 files changed

+2
-44
lines changed

2 files changed

+2
-44
lines changed

‎aten/src/ATen/native/cuda/ForeachUnaryOp.cu

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -237,7 +237,7 @@ void floating_half_bfloat16_(TensorList tensors) {
237237
OP_CUSTOM_FUNCTOR(function, op_name, functor_name);
238238

239239
OP(floating_half_bfloat16, erfc, Erfc);
240-
OP(floating_half, lgamma, Lgamma);
240+
OP(floating_half_bfloat16, lgamma, Lgamma);
241241
OP(floating_half_bfloat16, trunc, Truncf);
242242
OP(floating_half_bfloat16, floor, Floor);
243243
OP(floating_half_bfloat16, ceil, Ceil);

‎torch/testing/_internal/common_methods_invocations.py

Lines changed: 1 addition & 43 deletions
Original file line numberDiff line numberDiff line change
@@ -10798,84 +10798,42 @@ def __call__(self, opinfo, device, dtype, requires_grad, **kwargs):
1079810798
"test_dispatch_meta_inplace",
1079910799
dtypes=complex_types() + integral_types_and(torch.bool),
1080010800
),
10801-
DecorateInfo(
10802-
unittest.expectedFailure,
10803-
"TestMeta",
10804-
"test_dispatch_meta_inplace",
10805-
device_type="cuda",
10806-
dtypes=(torch.bfloat16,),
10807-
),
1080810801
DecorateInfo(
1080910802
unittest.expectedFailure,
1081010803
"TestMeta",
1081110804
"test_dispatch_meta_outplace",
1081210805
dtypes=complex_types(),
1081310806
),
10814-
DecorateInfo(
10815-
unittest.expectedFailure,
10816-
"TestMeta",
10817-
"test_dispatch_meta_outplace",
10818-
device_type="cuda",
10819-
dtypes=(torch.bfloat16,),
10820-
),
1082110807
DecorateInfo(
1082210808
unittest.expectedFailure,
1082310809
"TestMeta",
1082410810
"test_dispatch_symbolic_meta_inplace",
1082510811
dtypes=complex_types() + integral_types_and(torch.bool),
1082610812
),
10827-
DecorateInfo(
10828-
unittest.expectedFailure,
10829-
"TestMeta",
10830-
"test_dispatch_symbolic_meta_inplace",
10831-
device_type="cuda",
10832-
dtypes=(torch.bfloat16,),
10833-
),
1083410813
DecorateInfo(
1083510814
unittest.expectedFailure,
1083610815
"TestMeta",
1083710816
"test_dispatch_symbolic_meta_outplace",
1083810817
dtypes=complex_types(),
1083910818
),
10840-
DecorateInfo(
10841-
unittest.expectedFailure,
10842-
"TestMeta",
10843-
"test_dispatch_symbolic_meta_outplace",
10844-
device_type="cuda",
10845-
dtypes=(torch.bfloat16,),
10846-
),
1084710819
DecorateInfo(
1084810820
unittest.expectedFailure,
1084910821
"TestMeta",
1085010822
"test_meta_inplace",
1085110823
dtypes=complex_types() + integral_types_and(torch.bool),
1085210824
),
10853-
DecorateInfo(
10854-
unittest.expectedFailure,
10855-
"TestMeta",
10856-
"test_meta_inplace",
10857-
device_type="cuda",
10858-
dtypes=(torch.bfloat16,),
10859-
),
1086010825
DecorateInfo(
1086110826
unittest.expectedFailure,
1086210827
"TestMeta",
1086310828
"test_meta_outplace",
1086410829
dtypes=complex_types(),
1086510830
),
10866-
DecorateInfo(
10867-
unittest.expectedFailure,
10868-
"TestMeta",
10869-
"test_meta_outplace",
10870-
device_type="cuda",
10871-
dtypes=(torch.bfloat16,),
10872-
),
1087310831
DecorateInfo(
1087410832
unittest.expectedFailure,
1087510833
"TestForeach",
1087610834
"test_parity",
1087710835
device_type="cuda",
10878-
dtypes=complex_types() + (torch.bfloat16,),
10836+
dtypes=complex_types(),
1087910837
active_if=lambda kwargs: not kwargs.get("noncontiguous", False),
1088010838
),
1088110839
DecorateInfo(

0 commit comments

Comments
 (0)
0