Skip to content

Commit 6d20a9f

Browse files
committed
add NVJPEG error meassage spider to fix ci build. test=develop
1 parent 03783f2 commit 6d20a9f

File tree

2 files changed

+46
-1
lines changed

2 files changed

+46
-1
lines changed

paddle/fluid/operators/data/map_runner.h

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -14,7 +14,6 @@
1414
#include <memory>
1515
#include <mutex>
1616
#include <vector>
17-
#include "ThreadPool.h"
1817

1918
#include "paddle/fluid/framework/parallel_executor.h"
2019
#include "paddle/fluid/operators/reader/lod_tensor_blocking_queue.h"

tools/externalError/spider.py

Lines changed: 46 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -361,6 +361,52 @@ def handle_data(self, data):
361361
desc.strip())
362362

363363
CUFFTHTMLParser().feed(html)
364+
#*************************************************************************************************#
365+
366+
#*********************************** nvJPEG Error Message **************************************#
367+
nvjpegStatus_t = {
368+
"NVJPEG_STATUS_SUCCESS": 0,
369+
"NVJPEG_STATUS_NOT_INITIALIZED": 1,
370+
"NVJPEG_STATUS_INVALID_PARAMETER": 2,
371+
"NVJPEG_STATUS_BAD_JPEG": 3,
372+
"NVJPEG_STATUS_JPEG_NOT_SUPPORTED": 4,
373+
"NVJPEG_STATUS_ALLOCATOR_FAILURE": 5,
374+
"NVJPEG_STATUS_EXECUTION_FAILED": 6,
375+
"NVJPEG_STATUS_ARCH_MISMATCH": 7,
376+
"NVJPEG_STATUS_INTERNAL_ERROR": 8,
377+
"NVJPEG_STATUS_IMPLEMENTATION_NOT_SUPPORTED": 9,
378+
"NVJPEG_STATUS_INCOMPLETE_BITSTREAM": 10,
379+
}
380+
381+
print("start crawling errorMessage for nvidia nvJPEG API--->")
382+
url = 'https://docs.nvidia.com/cuda/nvjpeg/#nvjpeg-api-return-codes'
383+
384+
allMessageDesc = externalErrorDesc.errors.add()
385+
allMessageDesc.type = external_error_pb2.NVJPEG
386+
387+
html = urllib.request.urlopen(url).read().decode('utf-8')
388+
389+
res_div = r'Description of the returned error codes:.*?<div class="tablenoborder">(.*?)</div>'
390+
m_div = re.findall(res_div, html, re.S | re.M)[0]
391+
392+
res_dt = r'<samp class="ph codeph">(.*?)</samp></td>.*?colspan="1">(.*?)</td>'
393+
m_dt = re.findall(res_dt, m_div, re.S | re.M)
394+
395+
for error in m_dt:
396+
m_code = error[0]
397+
m_code = m_code.split()[0].strip()
398+
399+
m_message = error[1]
400+
m_message = re.sub(r'\t', ' ', m_message)
401+
m_message = re.sub(r'\n +', ' ', m_message)
402+
m_message = re.sub(r'<.*?>', '', m_message)
403+
404+
_Messages = allMessageDesc.messages.add()
405+
_Messages.code = int(nvjpegStatus_t[m_code])
406+
_Messages.message = "'%s'. %s" % (m_code, m_message)
407+
408+
print("End crawling errorMessage for nvidia NVJPEG API!\n")
409+
#*************************************************************************************************#
364410

365411

366412
def main(argv):

0 commit comments

Comments
 (0)