File tree Expand file tree Collapse file tree 4 files changed +9
-3
lines changed Expand file tree Collapse file tree 4 files changed +9
-3
lines changed Original file line number Diff line number Diff line change @@ -44,6 +44,9 @@ function(op_library TARGET)
44
44
if (EXISTS ${CMAKE_CURRENT_SOURCE_DIR} /${TARGET}.cu )
45
45
list (APPEND cu_srcs ${TARGET} .cu )
46
46
endif ()
47
+ if (WITH_NV_JETSON )
48
+ list (REMOVE_ITEM cu_srcs "decode_jpeg_op.cu" )
49
+ endif ()
47
50
if (EXISTS ${CMAKE_CURRENT_SOURCE_DIR} /${TARGET}.part.cu )
48
51
set (PART_CUDA_KERNEL_FILES ${CMAKE_CURRENT_SOURCE_DIR} /${TARGET}.part.cu
49
52
${PART_CUDA_KERNEL_FILES} PARENT_SCOPE )
Original file line number Diff line number Diff line change 19
19
#include " paddle/fluid/framework/generator.h"
20
20
#include " paddle/fluid/framework/op_registry.h"
21
21
#include " paddle/fluid/framework/operator.h"
22
- #include " paddle/fluid/platform/dynload/nvjpeg.h"
23
22
#include " paddle/fluid/platform/enforce.h"
24
23
25
24
namespace paddle {
Original file line number Diff line number Diff line change 12
12
// See the License for the specific language governing permissions and
13
13
// limitations under the License.
14
14
15
- #ifndef PADDLE_WITH_HIP
15
+ #if !defined(WITH_NV_JETSON) && !defined( PADDLE_WITH_HIP)
16
16
17
17
#include < string>
18
18
#include " paddle/fluid/framework/op_registry.h"
Original file line number Diff line number Diff line change 1
1
cc_library (dynamic_loader SRCS dynamic_loader.cc DEPS glog gflags enforce )
2
2
3
- list (APPEND CUDA_SRCS cublas.cc cudnn.cc curand.cc cusolver.cc nvtx.cc nvjpeg.cc )
3
+ list (APPEND CUDA_SRCS cublas.cc cudnn.cc curand.cc cusolver.cc nvtx.cc )
4
+
5
+ if (NOT WITH_NV_JETSON )
6
+ list (APPEND CUDA_SRCS nvjpeg.cc )
7
+ endif ()
4
8
5
9
if (WITH_ROCM )
6
10
list (APPEND HIP_SRCS rocblas.cc miopen.cc hiprand.cc )
You can’t perform that action at this time.
0 commit comments