Skip to content

Commit 57a4901

Browse files
cyyeverpytorchmergebot
cyyever
authored and
pytorchmergebot
committed
[5/N] Fix Wextra-semi warning (pytorch#139465)
Fixes #ISSUE_NUMBER Pull Request resolved: pytorch#139465 Approved by: https://github.com/ezyang
1 parent 03e8311 commit 57a4901

File tree

139 files changed

+405
-405
lines changed

Some content is hidden

Large Commits have some content hidden by default. Use the searchbox below for content that may be hidden.

139 files changed

+405
-405
lines changed

‎aten/src/ATen/ParallelThreadPoolNative.cpp

+1-1
Original file line numberDiff line numberDiff line change
@@ -45,7 +45,7 @@ std::shared_ptr<TaskThreadPoolBase> create_c10_threadpool(
4545

4646
} // namespace
4747

48-
C10_REGISTER_CREATOR(ThreadPoolRegistry, C10, create_c10_threadpool);
48+
C10_REGISTER_CREATOR(ThreadPoolRegistry, C10, create_c10_threadpool)
4949

5050
void set_num_interop_threads(int nthreads) {
5151
TORCH_CHECK(nthreads > 0, "Expected positive number of threads");

‎aten/src/ATen/cpu/vec/vec256/vec256_bfloat16.h

+2-2
Original file line numberDiff line numberDiff line change
@@ -1119,8 +1119,8 @@ inline void load_fp32_from_##name(const type *data, Vectorized<float>& out1, Vec
11191119
out1 = out1_values; \
11201120
out2 = out2_values; \
11211121
}
1122-
LOAD_FP32_VECTORIZED_INIT(BFloat16, bf16);
1123-
LOAD_FP32_VECTORIZED_INIT(Half, fp16);
1122+
LOAD_FP32_VECTORIZED_INIT(BFloat16, bf16)
1123+
LOAD_FP32_VECTORIZED_INIT(Half, fp16)
11241124

11251125
#else // defined(CPU_CAPABILITY_AVX2)
11261126
#define LOAD_FP32_NON_VECTORIZED_INIT(type, name) \

‎aten/src/ATen/cpu/vec/vec512/vec512_bfloat16.h

+2-2
Original file line numberDiff line numberDiff line change
@@ -1707,8 +1707,8 @@ inline void load_fp32_from_##name(const type *data, Vectorized<float>& out1, Vec
17071707
out1 = out1_values; \
17081708
out2 = out2_values; \
17091709
}
1710-
LOAD_FP32_VECTORIZED_INIT(BFloat16, bf16);
1711-
LOAD_FP32_VECTORIZED_INIT(Half, fp16);
1710+
LOAD_FP32_VECTORIZED_INIT(BFloat16, bf16)
1711+
LOAD_FP32_VECTORIZED_INIT(Half, fp16)
17121712

17131713
#else // defined(CPU_CAPABILITY_AVX512)
17141714
#define LOAD_FP32_NON_VECTORIZED_INIT(type, name) \

‎aten/src/ATen/hip/impl/HIPGuardImplMasqueradingAsCUDA.cpp

+1-1
Original file line numberDiff line numberDiff line change
@@ -11,4 +11,4 @@
1111
//
1212
// This hack can be removed once PyTorch is out-of-place HIPified, and
1313
// doesn't pretend CUDA is HIP.
14-
C10_REGISTER_GUARD_IMPL(CUDA, at::cuda::HIPGuardImplMasqueradingAsCUDA);
14+
C10_REGISTER_GUARD_IMPL(CUDA, at::cuda::HIPGuardImplMasqueradingAsCUDA)

‎aten/src/ATen/mps/MPSAllocatorInterface.h

+1-1
Original file line numberDiff line numberDiff line change
@@ -55,7 +55,7 @@ class IMpsAllocatorCallback {
5555
// MPS allocator will execute every registered callback when a block of memory is freed.
5656
C10_DECLARE_REGISTRY(MPSAllocatorCallbacksRegistry, IMpsAllocatorCallback);
5757
#define REGISTER_MPS_ALLOCATOR_CALLBACK(name, ...) \
58-
C10_REGISTER_CLASS(MPSAllocatorCallbacksRegistry, name, __VA_ARGS__);
58+
C10_REGISTER_CLASS(MPSAllocatorCallbacksRegistry, name, __VA_ARGS__)
5959

6060
IMPSAllocator* getIMPSAllocator(bool sharedAllocator = false);
6161

‎aten/src/ATen/mps/MPSGuardImpl.h

+1-1
Original file line numberDiff line numberDiff line change
@@ -176,6 +176,6 @@ struct OptionalMPSGuard {
176176
};
177177

178178

179-
C10_REGISTER_GUARD_IMPL(MPS, MPSGuardImpl);
179+
C10_REGISTER_GUARD_IMPL(MPS, MPSGuardImpl)
180180

181181
} // namespace at::mps

‎aten/src/ATen/native/BatchLinearAlgebraKernel.cpp

+104-104
Original file line numberDiff line numberDiff line change
@@ -1135,108 +1135,108 @@ void unpack_pivots_cpu_kernel(TensorIterator& iter, const int64_t dim_size, cons
11351135
}
11361136
} // anonymous namespace
11371137

1138-
REGISTER_ARCH_DISPATCH(cholesky_stub, DEFAULT, &cholesky_kernel);
1139-
REGISTER_AVX512_DISPATCH(cholesky_stub, &cholesky_kernel);
1140-
REGISTER_AVX2_DISPATCH(cholesky_stub, &cholesky_kernel);
1141-
REGISTER_VSX_DISPATCH(cholesky_stub, &cholesky_kernel);
1142-
REGISTER_ZVECTOR_DISPATCH(cholesky_stub, &cholesky_kernel);
1143-
REGISTER_SVE256_DISPATCH(cholesky_stub, &cholesky_kernel);
1144-
1145-
REGISTER_ARCH_DISPATCH(cholesky_inverse_stub, DEFAULT, &cholesky_inverse_kernel_impl);
1146-
REGISTER_AVX512_DISPATCH(cholesky_inverse_stub, &cholesky_inverse_kernel_impl);
1147-
REGISTER_AVX2_DISPATCH(cholesky_inverse_stub, &cholesky_inverse_kernel_impl);
1148-
REGISTER_VSX_DISPATCH(cholesky_inverse_stub, &cholesky_inverse_kernel_impl);
1149-
REGISTER_ZVECTOR_DISPATCH(cholesky_inverse_stub, &cholesky_inverse_kernel_impl);
1150-
REGISTER_SVE256_DISPATCH(cholesky_inverse_stub, &cholesky_inverse_kernel_impl);
1151-
1152-
REGISTER_ARCH_DISPATCH(linalg_eig_stub, DEFAULT, &linalg_eig_kernel);
1153-
REGISTER_AVX512_DISPATCH(linalg_eig_stub, &linalg_eig_kernel);
1154-
REGISTER_AVX2_DISPATCH(linalg_eig_stub, &linalg_eig_kernel);
1155-
REGISTER_VSX_DISPATCH(linalg_eig_stub, &linalg_eig_kernel);
1156-
REGISTER_ZVECTOR_DISPATCH(linalg_eig_stub, &linalg_eig_kernel);
1157-
REGISTER_SVE256_DISPATCH(linalg_eig_stub, &linalg_eig_kernel);
1158-
1159-
REGISTER_ARCH_DISPATCH(linalg_eigh_stub, DEFAULT, &linalg_eigh_kernel);
1160-
REGISTER_AVX512_DISPATCH(linalg_eigh_stub, &linalg_eigh_kernel);
1161-
REGISTER_AVX2_DISPATCH(linalg_eigh_stub, &linalg_eigh_kernel);
1162-
REGISTER_VSX_DISPATCH(linalg_eigh_stub, &linalg_eigh_kernel);
1163-
REGISTER_ZVECTOR_DISPATCH(linalg_eigh_stub, &linalg_eigh_kernel);
1164-
REGISTER_SVE256_DISPATCH(linalg_eigh_stub, &linalg_eigh_kernel);
1165-
1166-
REGISTER_ARCH_DISPATCH(geqrf_stub, DEFAULT, &geqrf_kernel);
1167-
REGISTER_AVX512_DISPATCH(geqrf_stub, &geqrf_kernel);
1168-
REGISTER_AVX2_DISPATCH(geqrf_stub, &geqrf_kernel);
1169-
REGISTER_VSX_DISPATCH(geqrf_stub, &geqrf_kernel);
1170-
REGISTER_ZVECTOR_DISPATCH(geqrf_stub, &geqrf_kernel);
1171-
REGISTER_SVE256_DISPATCH(geqrf_stub, &geqrf_kernel);
1172-
1173-
REGISTER_ARCH_DISPATCH(orgqr_stub, DEFAULT, &orgqr_kernel_impl);
1174-
REGISTER_AVX512_DISPATCH(orgqr_stub, &orgqr_kernel_impl);
1175-
REGISTER_AVX2_DISPATCH(orgqr_stub, &orgqr_kernel_impl);
1176-
REGISTER_VSX_DISPATCH(orgqr_stub, &orgqr_kernel_impl);
1177-
REGISTER_ZVECTOR_DISPATCH(orgqr_stub, &orgqr_kernel_impl);
1178-
REGISTER_SVE256_DISPATCH(orgqr_stub, &orgqr_kernel_impl);
1179-
1180-
REGISTER_ARCH_DISPATCH(ormqr_stub, DEFAULT, &ormqr_kernel);
1181-
REGISTER_AVX512_DISPATCH(ormqr_stub, &ormqr_kernel);
1182-
REGISTER_AVX2_DISPATCH(ormqr_stub, &ormqr_kernel);
1183-
REGISTER_VSX_DISPATCH(ormqr_stub, &ormqr_kernel);
1184-
REGISTER_ZVECTOR_DISPATCH(ormqr_stub, &ormqr_kernel);
1185-
REGISTER_SVE256_DISPATCH(ormqr_stub, &ormqr_kernel);
1186-
1187-
REGISTER_ARCH_DISPATCH(lstsq_stub, DEFAULT, &lstsq_kernel);
1188-
REGISTER_AVX512_DISPATCH(lstsq_stub, &lstsq_kernel);
1189-
REGISTER_AVX2_DISPATCH(lstsq_stub, &lstsq_kernel);
1190-
REGISTER_VSX_DISPATCH(lstsq_stub, &lstsq_kernel);
1191-
REGISTER_ZVECTOR_DISPATCH(lstsq_stub, &lstsq_kernel);
1192-
REGISTER_SVE256_DISPATCH(lstsq_stub, &lstsq_kernel);
1193-
1194-
REGISTER_ARCH_DISPATCH(triangular_solve_stub, DEFAULT, &triangular_solve_kernel);
1195-
REGISTER_AVX512_DISPATCH(triangular_solve_stub, &triangular_solve_kernel);
1196-
REGISTER_AVX2_DISPATCH(triangular_solve_stub, &triangular_solve_kernel);
1197-
REGISTER_VSX_DISPATCH(triangular_solve_stub, &triangular_solve_kernel);
1198-
REGISTER_ZVECTOR_DISPATCH(triangular_solve_stub, &triangular_solve_kernel);
1199-
REGISTER_SVE256_DISPATCH(triangular_solve_stub, &triangular_solve_kernel);
1200-
1201-
REGISTER_ARCH_DISPATCH(lu_factor_stub, DEFAULT, &lu_factor_kernel);
1202-
REGISTER_AVX512_DISPATCH(lu_factor_stub, &lu_factor_kernel);
1203-
REGISTER_AVX2_DISPATCH(lu_factor_stub, &lu_factor_kernel);
1204-
REGISTER_VSX_DISPATCH(lu_factor_stub, &lu_factor_kernel);
1205-
REGISTER_ZVECTOR_DISPATCH(lu_factor_stub, &lu_factor_kernel);
1206-
REGISTER_SVE256_DISPATCH(lu_factor_stub, &lu_factor_kernel);
1207-
1208-
REGISTER_ARCH_DISPATCH(ldl_factor_stub, DEFAULT, &ldl_factor_kernel);
1209-
REGISTER_AVX512_DISPATCH(ldl_factor_stub, &ldl_factor_kernel);
1210-
REGISTER_AVX2_DISPATCH(ldl_factor_stub, &ldl_factor_kernel);
1211-
REGISTER_VSX_DISPATCH(ldl_factor_stub, &ldl_factor_kernel);
1212-
REGISTER_ZVECTOR_DISPATCH(ldl_factor_stub, &ldl_factor_kernel);
1213-
REGISTER_SVE256_DISPATCH(ldl_factor_stub, &ldl_factor_kernel);
1214-
1215-
REGISTER_ARCH_DISPATCH(ldl_solve_stub, DEFAULT, &ldl_solve_kernel);
1216-
REGISTER_AVX512_DISPATCH(ldl_solve_stub, &ldl_solve_kernel);
1217-
REGISTER_AVX2_DISPATCH(ldl_solve_stub, &ldl_solve_kernel);
1218-
REGISTER_VSX_DISPATCH(ldl_solve_stub, &ldl_solve_kernel);
1219-
REGISTER_ZVECTOR_DISPATCH(ldl_solve_stub, &ldl_solve_kernel);
1220-
REGISTER_SVE256_DISPATCH(ldl_solve_stub, &ldl_solve_kernel);
1221-
1222-
REGISTER_ARCH_DISPATCH(lu_solve_stub, DEFAULT, &lu_solve_kernel);
1223-
REGISTER_AVX512_DISPATCH(lu_solve_stub, &lu_solve_kernel);
1224-
REGISTER_AVX2_DISPATCH(lu_solve_stub, &lu_solve_kernel);
1225-
REGISTER_VSX_DISPATCH(lu_solve_stub, &lu_solve_kernel);
1226-
REGISTER_ZVECTOR_DISPATCH(lu_solve_stub, &lu_solve_kernel);
1227-
REGISTER_SVE256_DISPATCH(lu_solve_stub, &lu_solve_kernel);
1228-
1229-
REGISTER_ARCH_DISPATCH(svd_stub, DEFAULT, &svd_kernel);
1230-
REGISTER_AVX512_DISPATCH(svd_stub, &svd_kernel);
1231-
REGISTER_AVX2_DISPATCH(svd_stub, &svd_kernel);
1232-
REGISTER_VSX_DISPATCH(svd_stub, &svd_kernel);
1233-
REGISTER_ZVECTOR_DISPATCH(svd_stub, &svd_kernel);
1234-
REGISTER_SVE256_DISPATCH(svd_stub, &svd_kernel);
1235-
1236-
REGISTER_ARCH_DISPATCH(unpack_pivots_stub, DEFAULT, &unpack_pivots_cpu_kernel);
1237-
REGISTER_AVX512_DISPATCH(unpack_pivots_stub, &unpack_pivots_cpu_kernel);
1238-
REGISTER_AVX2_DISPATCH(unpack_pivots_stub, &unpack_pivots_cpu_kernel);
1239-
REGISTER_VSX_DISPATCH(unpack_pivots_stub, &unpack_pivots_cpu_kernel);
1240-
REGISTER_ZVECTOR_DISPATCH(unpack_pivots_stub, &unpack_pivots_cpu_kernel);
1241-
REGISTER_SVE256_DISPATCH(unpack_pivots_stub, &unpack_pivots_cpu_kernel);
1138+
REGISTER_ARCH_DISPATCH(cholesky_stub, DEFAULT, &cholesky_kernel)
1139+
REGISTER_AVX512_DISPATCH(cholesky_stub, &cholesky_kernel)
1140+
REGISTER_AVX2_DISPATCH(cholesky_stub, &cholesky_kernel)
1141+
REGISTER_VSX_DISPATCH(cholesky_stub, &cholesky_kernel)
1142+
REGISTER_ZVECTOR_DISPATCH(cholesky_stub, &cholesky_kernel)
1143+
REGISTER_SVE256_DISPATCH(cholesky_stub, &cholesky_kernel)
1144+
1145+
REGISTER_ARCH_DISPATCH(cholesky_inverse_stub, DEFAULT, &cholesky_inverse_kernel_impl)
1146+
REGISTER_AVX512_DISPATCH(cholesky_inverse_stub, &cholesky_inverse_kernel_impl)
1147+
REGISTER_AVX2_DISPATCH(cholesky_inverse_stub, &cholesky_inverse_kernel_impl)
1148+
REGISTER_VSX_DISPATCH(cholesky_inverse_stub, &cholesky_inverse_kernel_impl)
1149+
REGISTER_ZVECTOR_DISPATCH(cholesky_inverse_stub, &cholesky_inverse_kernel_impl)
1150+
REGISTER_SVE256_DISPATCH(cholesky_inverse_stub, &cholesky_inverse_kernel_impl)
1151+
1152+
REGISTER_ARCH_DISPATCH(linalg_eig_stub, DEFAULT, &linalg_eig_kernel)
1153+
REGISTER_AVX512_DISPATCH(linalg_eig_stub, &linalg_eig_kernel)
1154+
REGISTER_AVX2_DISPATCH(linalg_eig_stub, &linalg_eig_kernel)
1155+
REGISTER_VSX_DISPATCH(linalg_eig_stub, &linalg_eig_kernel)
1156+
REGISTER_ZVECTOR_DISPATCH(linalg_eig_stub, &linalg_eig_kernel)
1157+
REGISTER_SVE256_DISPATCH(linalg_eig_stub, &linalg_eig_kernel)
1158+
1159+
REGISTER_ARCH_DISPATCH(linalg_eigh_stub, DEFAULT, &linalg_eigh_kernel)
1160+
REGISTER_AVX512_DISPATCH(linalg_eigh_stub, &linalg_eigh_kernel)
1161+
REGISTER_AVX2_DISPATCH(linalg_eigh_stub, &linalg_eigh_kernel)
1162+
REGISTER_VSX_DISPATCH(linalg_eigh_stub, &linalg_eigh_kernel)
1163+
REGISTER_ZVECTOR_DISPATCH(linalg_eigh_stub, &linalg_eigh_kernel)
1164+
REGISTER_SVE256_DISPATCH(linalg_eigh_stub, &linalg_eigh_kernel)
1165+
1166+
REGISTER_ARCH_DISPATCH(geqrf_stub, DEFAULT, &geqrf_kernel)
1167+
REGISTER_AVX512_DISPATCH(geqrf_stub, &geqrf_kernel)
1168+
REGISTER_AVX2_DISPATCH(geqrf_stub, &geqrf_kernel)
1169+
REGISTER_VSX_DISPATCH(geqrf_stub, &geqrf_kernel)
1170+
REGISTER_ZVECTOR_DISPATCH(geqrf_stub, &geqrf_kernel)
1171+
REGISTER_SVE256_DISPATCH(geqrf_stub, &geqrf_kernel)
1172+
1173+
REGISTER_ARCH_DISPATCH(orgqr_stub, DEFAULT, &orgqr_kernel_impl)
1174+
REGISTER_AVX512_DISPATCH(orgqr_stub, &orgqr_kernel_impl)
1175+
REGISTER_AVX2_DISPATCH(orgqr_stub, &orgqr_kernel_impl)
1176+
REGISTER_VSX_DISPATCH(orgqr_stub, &orgqr_kernel_impl)
1177+
REGISTER_ZVECTOR_DISPATCH(orgqr_stub, &orgqr_kernel_impl)
1178+
REGISTER_SVE256_DISPATCH(orgqr_stub, &orgqr_kernel_impl)
1179+
1180+
REGISTER_ARCH_DISPATCH(ormqr_stub, DEFAULT, &ormqr_kernel)
1181+
REGISTER_AVX512_DISPATCH(ormqr_stub, &ormqr_kernel)
1182+
REGISTER_AVX2_DISPATCH(ormqr_stub, &ormqr_kernel)
1183+
REGISTER_VSX_DISPATCH(ormqr_stub, &ormqr_kernel)
1184+
REGISTER_ZVECTOR_DISPATCH(ormqr_stub, &ormqr_kernel)
1185+
REGISTER_SVE256_DISPATCH(ormqr_stub, &ormqr_kernel)
1186+
1187+
REGISTER_ARCH_DISPATCH(lstsq_stub, DEFAULT, &lstsq_kernel)
1188+
REGISTER_AVX512_DISPATCH(lstsq_stub, &lstsq_kernel)
1189+
REGISTER_AVX2_DISPATCH(lstsq_stub, &lstsq_kernel)
1190+
REGISTER_VSX_DISPATCH(lstsq_stub, &lstsq_kernel)
1191+
REGISTER_ZVECTOR_DISPATCH(lstsq_stub, &lstsq_kernel)
1192+
REGISTER_SVE256_DISPATCH(lstsq_stub, &lstsq_kernel)
1193+
1194+
REGISTER_ARCH_DISPATCH(triangular_solve_stub, DEFAULT, &triangular_solve_kernel)
1195+
REGISTER_AVX512_DISPATCH(triangular_solve_stub, &triangular_solve_kernel)
1196+
REGISTER_AVX2_DISPATCH(triangular_solve_stub, &triangular_solve_kernel)
1197+
REGISTER_VSX_DISPATCH(triangular_solve_stub, &triangular_solve_kernel)
1198+
REGISTER_ZVECTOR_DISPATCH(triangular_solve_stub, &triangular_solve_kernel)
1199+
REGISTER_SVE256_DISPATCH(triangular_solve_stub, &triangular_solve_kernel)
1200+
1201+
REGISTER_ARCH_DISPATCH(lu_factor_stub, DEFAULT, &lu_factor_kernel)
1202+
REGISTER_AVX512_DISPATCH(lu_factor_stub, &lu_factor_kernel)
1203+
REGISTER_AVX2_DISPATCH(lu_factor_stub, &lu_factor_kernel)
1204+
REGISTER_VSX_DISPATCH(lu_factor_stub, &lu_factor_kernel)
1205+
REGISTER_ZVECTOR_DISPATCH(lu_factor_stub, &lu_factor_kernel)
1206+
REGISTER_SVE256_DISPATCH(lu_factor_stub, &lu_factor_kernel)
1207+
1208+
REGISTER_ARCH_DISPATCH(ldl_factor_stub, DEFAULT, &ldl_factor_kernel)
1209+
REGISTER_AVX512_DISPATCH(ldl_factor_stub, &ldl_factor_kernel)
1210+
REGISTER_AVX2_DISPATCH(ldl_factor_stub, &ldl_factor_kernel)
1211+
REGISTER_VSX_DISPATCH(ldl_factor_stub, &ldl_factor_kernel)
1212+
REGISTER_ZVECTOR_DISPATCH(ldl_factor_stub, &ldl_factor_kernel)
1213+
REGISTER_SVE256_DISPATCH(ldl_factor_stub, &ldl_factor_kernel)
1214+
1215+
REGISTER_ARCH_DISPATCH(ldl_solve_stub, DEFAULT, &ldl_solve_kernel)
1216+
REGISTER_AVX512_DISPATCH(ldl_solve_stub, &ldl_solve_kernel)
1217+
REGISTER_AVX2_DISPATCH(ldl_solve_stub, &ldl_solve_kernel)
1218+
REGISTER_VSX_DISPATCH(ldl_solve_stub, &ldl_solve_kernel)
1219+
REGISTER_ZVECTOR_DISPATCH(ldl_solve_stub, &ldl_solve_kernel)
1220+
REGISTER_SVE256_DISPATCH(ldl_solve_stub, &ldl_solve_kernel)
1221+
1222+
REGISTER_ARCH_DISPATCH(lu_solve_stub, DEFAULT, &lu_solve_kernel)
1223+
REGISTER_AVX512_DISPATCH(lu_solve_stub, &lu_solve_kernel)
1224+
REGISTER_AVX2_DISPATCH(lu_solve_stub, &lu_solve_kernel)
1225+
REGISTER_VSX_DISPATCH(lu_solve_stub, &lu_solve_kernel)
1226+
REGISTER_ZVECTOR_DISPATCH(lu_solve_stub, &lu_solve_kernel)
1227+
REGISTER_SVE256_DISPATCH(lu_solve_stub, &lu_solve_kernel)
1228+
1229+
REGISTER_ARCH_DISPATCH(svd_stub, DEFAULT, &svd_kernel)
1230+
REGISTER_AVX512_DISPATCH(svd_stub, &svd_kernel)
1231+
REGISTER_AVX2_DISPATCH(svd_stub, &svd_kernel)
1232+
REGISTER_VSX_DISPATCH(svd_stub, &svd_kernel)
1233+
REGISTER_ZVECTOR_DISPATCH(svd_stub, &svd_kernel)
1234+
REGISTER_SVE256_DISPATCH(svd_stub, &svd_kernel)
1235+
1236+
REGISTER_ARCH_DISPATCH(unpack_pivots_stub, DEFAULT, &unpack_pivots_cpu_kernel)
1237+
REGISTER_AVX512_DISPATCH(unpack_pivots_stub, &unpack_pivots_cpu_kernel)
1238+
REGISTER_AVX2_DISPATCH(unpack_pivots_stub, &unpack_pivots_cpu_kernel)
1239+
REGISTER_VSX_DISPATCH(unpack_pivots_stub, &unpack_pivots_cpu_kernel)
1240+
REGISTER_ZVECTOR_DISPATCH(unpack_pivots_stub, &unpack_pivots_cpu_kernel)
1241+
REGISTER_SVE256_DISPATCH(unpack_pivots_stub, &unpack_pivots_cpu_kernel)
12421242
} // namespace at::native

‎aten/src/ATen/native/QuantizedLinear.cpp

+1-1
Original file line numberDiff line numberDiff line change
@@ -32,7 +32,7 @@
3232
#endif // USE_FBGEMM
3333

3434
namespace caffe2 {
35-
CAFFE_KNOWN_TYPE(c10::intrusive_ptr<LinearPackedParamsBase>);
35+
CAFFE_KNOWN_TYPE(c10::intrusive_ptr<LinearPackedParamsBase>)
3636
} // namespace caffe2
3737

3838
#ifdef USE_FBGEMM

‎aten/src/ATen/native/cpu/Activation.cpp

+17-17
Original file line numberDiff line numberDiff line change
@@ -1412,22 +1412,22 @@ REGISTER_DISPATCH(hardshrink_stub, &hardshrink_kernel)
14121412
REGISTER_DISPATCH(softshrink_stub, &softshrink_kernel)
14131413
REGISTER_DISPATCH(shrink_backward_stub, &shrink_backward_kernel)
14141414

1415-
ALSO_REGISTER_AVX512_DISPATCH(log_sigmoid_cpu_stub, &log_sigmoid_cpu_kernel);
1416-
ALSO_REGISTER_AVX512_DISPATCH(log_sigmoid_backward_stub, &log_sigmoid_backward_cpu_kernel);
1417-
ALSO_REGISTER_AVX512_DISPATCH(glu_stub, &glu_kernel);
1418-
ALSO_REGISTER_AVX512_DISPATCH(glu_backward_stub, &glu_backward_kernel);
1419-
ALSO_REGISTER_AVX512_DISPATCH(glu_jvp_stub, &glu_jvp_kernel);
1420-
ALSO_REGISTER_AVX512_DISPATCH(elu_stub, &elu_kernel);
1421-
ALSO_REGISTER_AVX512_DISPATCH(elu_backward_stub, &elu_backward_kernel);
1422-
ALSO_REGISTER_AVX512_DISPATCH(GeluKernel, &GeluKernelImpl);
1423-
ALSO_REGISTER_AVX512_DISPATCH(GeluBackwardKernel, &GeluBackwardKernelImpl);
1424-
ALSO_REGISTER_AVX512_DISPATCH(hardswish_stub, &hardswish_kernel);
1425-
ALSO_REGISTER_AVX512_DISPATCH(hardswish_backward_stub, &hardswish_backward_kernel);
1426-
ALSO_REGISTER_AVX512_DISPATCH(softplus_stub, &softplus_kernel);
1427-
ALSO_REGISTER_AVX512_DISPATCH(softplus_backward_stub, &softplus_backward_kernel);
1428-
ALSO_REGISTER_AVX512_DISPATCH(silu_stub, &silu_kernel);
1429-
ALSO_REGISTER_AVX512_DISPATCH(silu_backward_stub, &silu_backward_kernel);
1430-
ALSO_REGISTER_AVX512_DISPATCH(mish_stub, &mish_kernel);
1431-
ALSO_REGISTER_AVX512_DISPATCH(mish_backward_stub, &mish_backward_kernel);
1415+
ALSO_REGISTER_AVX512_DISPATCH(log_sigmoid_cpu_stub, &log_sigmoid_cpu_kernel)
1416+
ALSO_REGISTER_AVX512_DISPATCH(log_sigmoid_backward_stub, &log_sigmoid_backward_cpu_kernel)
1417+
ALSO_REGISTER_AVX512_DISPATCH(glu_stub, &glu_kernel)
1418+
ALSO_REGISTER_AVX512_DISPATCH(glu_backward_stub, &glu_backward_kernel)
1419+
ALSO_REGISTER_AVX512_DISPATCH(glu_jvp_stub, &glu_jvp_kernel)
1420+
ALSO_REGISTER_AVX512_DISPATCH(elu_stub, &elu_kernel)
1421+
ALSO_REGISTER_AVX512_DISPATCH(elu_backward_stub, &elu_backward_kernel)
1422+
ALSO_REGISTER_AVX512_DISPATCH(GeluKernel, &GeluKernelImpl)
1423+
ALSO_REGISTER_AVX512_DISPATCH(GeluBackwardKernel, &GeluBackwardKernelImpl)
1424+
ALSO_REGISTER_AVX512_DISPATCH(hardswish_stub, &hardswish_kernel)
1425+
ALSO_REGISTER_AVX512_DISPATCH(hardswish_backward_stub, &hardswish_backward_kernel)
1426+
ALSO_REGISTER_AVX512_DISPATCH(softplus_stub, &softplus_kernel)
1427+
ALSO_REGISTER_AVX512_DISPATCH(softplus_backward_stub, &softplus_backward_kernel)
1428+
ALSO_REGISTER_AVX512_DISPATCH(silu_stub, &silu_kernel)
1429+
ALSO_REGISTER_AVX512_DISPATCH(silu_backward_stub, &silu_backward_kernel)
1430+
ALSO_REGISTER_AVX512_DISPATCH(mish_stub, &mish_kernel)
1431+
ALSO_REGISTER_AVX512_DISPATCH(mish_backward_stub, &mish_backward_kernel)
14321432

14331433
} // namespace at::native

‎aten/src/ATen/native/cpu/BinaryOpsKernel.cpp

+12-12
Original file line numberDiff line numberDiff line change
@@ -1430,17 +1430,17 @@ REGISTER_DISPATCH(chebyshev_polynomial_u_stub, &chebyshev_polynomial_u_kernel)
14301430
REGISTER_DISPATCH(hermite_polynomial_h_stub, &hermite_polynomial_h_kernel)
14311431
REGISTER_DISPATCH(hermite_polynomial_he_stub, &hermite_polynomial_he_kernel)
14321432

1433-
ALSO_REGISTER_AVX512_DISPATCH(atan2_stub, &atan2_kernel);
1434-
ALSO_REGISTER_AVX512_DISPATCH(smooth_l1_stub, &smooth_l1_kernel);
1435-
ALSO_REGISTER_AVX512_DISPATCH(huber_stub, &huber_kernel);
1436-
ALSO_REGISTER_AVX512_DISPATCH(sigmoid_backward_stub, &sigmoid_backward_kernel);
1437-
ALSO_REGISTER_AVX512_DISPATCH(logit_backward_stub, &logit_backward_kernel);
1438-
ALSO_REGISTER_AVX512_DISPATCH(tanh_backward_stub, &tanh_backward_kernel);
1439-
ALSO_REGISTER_AVX512_DISPATCH(mse_stub, &mse_kernel);
1440-
ALSO_REGISTER_AVX512_DISPATCH(logaddexp_stub, &logaddexp_kernel);
1441-
ALSO_REGISTER_AVX512_DISPATCH(logaddexp2_stub, &logaddexp2_kernel);
1442-
ALSO_REGISTER_AVX512_DISPATCH(hypot_stub, &hypot_kernel);
1443-
ALSO_REGISTER_AVX512_DISPATCH(igamma_stub, &igamma_kernel);
1444-
ALSO_REGISTER_AVX512_DISPATCH(igammac_stub, &igammac_kernel);
1433+
ALSO_REGISTER_AVX512_DISPATCH(atan2_stub, &atan2_kernel)
1434+
ALSO_REGISTER_AVX512_DISPATCH(smooth_l1_stub, &smooth_l1_kernel)
1435+
ALSO_REGISTER_AVX512_DISPATCH(huber_stub, &huber_kernel)
1436+
ALSO_REGISTER_AVX512_DISPATCH(sigmoid_backward_stub, &sigmoid_backward_kernel)
1437+
ALSO_REGISTER_AVX512_DISPATCH(logit_backward_stub, &logit_backward_kernel)
1438+
ALSO_REGISTER_AVX512_DISPATCH(tanh_backward_stub, &tanh_backward_kernel)
1439+
ALSO_REGISTER_AVX512_DISPATCH(mse_stub, &mse_kernel)
1440+
ALSO_REGISTER_AVX512_DISPATCH(logaddexp_stub, &logaddexp_kernel)
1441+
ALSO_REGISTER_AVX512_DISPATCH(logaddexp2_stub, &logaddexp2_kernel)
1442+
ALSO_REGISTER_AVX512_DISPATCH(hypot_stub, &hypot_kernel)
1443+
ALSO_REGISTER_AVX512_DISPATCH(igamma_stub, &igamma_kernel)
1444+
ALSO_REGISTER_AVX512_DISPATCH(igammac_stub, &igammac_kernel)
14451445

14461446
} // namespace at::native

‎aten/src/ATen/native/cpu/ComplexKernel.cpp

+1-1
Original file line numberDiff line numberDiff line change
@@ -26,6 +26,6 @@ void polar_kernel(TensorIterator& iter) {
2626
} // anonymous namespace
2727

2828
REGISTER_DISPATCH(complex_stub, &complex_kernel)
29-
ALSO_REGISTER_AVX512_DISPATCH(polar_stub, &polar_kernel);
29+
ALSO_REGISTER_AVX512_DISPATCH(polar_stub, &polar_kernel)
3030

3131
} // namespace at::native

‎aten/src/ATen/native/cpu/DepthwiseConvKernel.cpp

+1-1
Original file line numberDiff line numberDiff line change
@@ -520,6 +520,6 @@ Tensor _convolution_depthwise3x3_winograd(
520520

521521
} // namespace
522522

523-
ALSO_REGISTER_AVX512_DISPATCH(convolution_depthwise3x3_winograd_stub, &_convolution_depthwise3x3_winograd);
523+
ALSO_REGISTER_AVX512_DISPATCH(convolution_depthwise3x3_winograd_stub, &_convolution_depthwise3x3_winograd)
524524

525525
} // namespace at::native

‎aten/src/ATen/native/cpu/FlashAttentionKernel.cpp

+2-2
Original file line numberDiff line numberDiff line change
@@ -1265,7 +1265,7 @@ void flash_attention_backward_kernel_impl(
12651265

12661266
} // anonymous namespace
12671267

1268-
ALSO_REGISTER_AVX512_DISPATCH(flash_attention_kernel, &flash_attention_kernel_impl);
1269-
ALSO_REGISTER_AVX512_DISPATCH(flash_attention_backward_kernel, &flash_attention_backward_kernel_impl);
1268+
ALSO_REGISTER_AVX512_DISPATCH(flash_attention_kernel, &flash_attention_kernel_impl)
1269+
ALSO_REGISTER_AVX512_DISPATCH(flash_attention_backward_kernel, &flash_attention_backward_kernel_impl)
12701270

12711271
} // at::native

‎aten/src/ATen/native/cpu/PowKernel.cpp

+2-2
Original file line numberDiff line numberDiff line change
@@ -144,7 +144,7 @@ static void pow_tensor_scalar_kernel(
144144

145145
} // anonymous namespace
146146

147-
ALSO_REGISTER_AVX512_DISPATCH(pow_tensor_tensor_stub, &CPU_CAPABILITY::pow_tensor_tensor_kernel);
148-
ALSO_REGISTER_AVX512_DISPATCH(pow_tensor_scalar_stub, &CPU_CAPABILITY::pow_tensor_scalar_kernel);
147+
ALSO_REGISTER_AVX512_DISPATCH(pow_tensor_tensor_stub, &CPU_CAPABILITY::pow_tensor_tensor_kernel)
148+
ALSO_REGISTER_AVX512_DISPATCH(pow_tensor_scalar_stub, &CPU_CAPABILITY::pow_tensor_scalar_kernel)
149149

150150
} // namespace at::native

0 commit comments

Comments
 (0)