azaion/annotations
1
0
Fork 0
mirror of https://github.com/azaion/annotations.git synced 2026-04-22 14:16:31 +00:00
Code Issues Packages Projects Releases Wiki Activity

fixed console Log

Browse Source 
fix same files problem in python different libs
correct command logging in command handler
This commit is contained in:
Alex Bezdieniezhnykh
2025-06-14 21:01:32 +03:00
parent 09cfcdf61a
commit c0f8dd792d
29 changed files with 74 additions and 87 deletions
Download Patch File Download Diff File Expand all files Collapse all files
Azaion.Inference/tensorrt_engine.pyx
+5 -5
View File
@@ -4,7 +4,7 @@ import pycuda.driver as cuda
import pycuda.autoinit # required for automatically initialize CUDA, do not remove.
import pynvml
import numpy as np
cimport constants
cimport constants_inf
cdef class TensorRTEngine(InferenceEngine):
@@ -100,16 +100,16 @@ cdef class TensorRTEngine(InferenceEngine):
return None
if builder.platform_has_fast_fp16:
constants.log('Converting to supported fp16')
constants_inf.log('Converting to supported fp16')
config.set_flag(trt.BuilderFlag.FP16)
else:
constants.log('Converting to supported fp32. (fp16 is not supported)')
constants_inf.log('Converting to supported fp32. (fp16 is not supported)')
plan = builder.build_serialized_network(network, config)
if plan is None:
constants.logerror('Conversion failed.')
constants_inf.logerror('Conversion failed.')
return None
constants.log('conversion done!')
constants_inf.log('conversion done!')
return bytes(plan)
cpdef tuple get_input_shape(self):