if(WITH_IPU)
  py_test(test_custom_nllloss_ipu SRCS test_custom_nllloss_ipu.py)
endif()
