pytorch/modules/detectron
cyy afd7b581aa Simplify OpenMP detection in CMake (#91576)
We greatly simplify the handing of OpenMP in CMake by using caffe2::openmp target thoroughly. We follow the old behavior by defaulting to MKL OMP library and detecting OMP flags otherwise.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/91576
Approved by: https://github.com/malfet
2023-02-04 11:50:06 +00:00
..
CMakeLists.txt
group_spatial_softmax_op.cc
group_spatial_softmax_op.cu
group_spatial_softmax_op.h
ps_roi_pool_op.cc
ps_roi_pool_op.cu
ps_roi_pool_op.h
roi_pool_f_op.cc
roi_pool_f_op.cu
roi_pool_f_op.h
sample_as_op.cc
sample_as_op.cu
sample_as_op.h
select_smooth_l1_loss_op.cc
select_smooth_l1_loss_op.cu
select_smooth_l1_loss_op.h
sigmoid_cross_entropy_loss_op.cc
sigmoid_cross_entropy_loss_op.cu
sigmoid_cross_entropy_loss_op.h
sigmoid_focal_loss_op.cc
sigmoid_focal_loss_op.cu
sigmoid_focal_loss_op.h
smooth_l1_loss_op.cc
smooth_l1_loss_op.cu
smooth_l1_loss_op.h
softmax_focal_loss_op.cc
softmax_focal_loss_op.cu
softmax_focal_loss_op.h
spatial_narrow_as_op.cc
spatial_narrow_as_op.cu
spatial_narrow_as_op.h
upsample_nearest_op.cc
upsample_nearest_op.cu
upsample_nearest_op.h
upsample_nearest_op_test.py