github/DeepFaceLive
1
0
Fork 0
mirror of https://github.com/iperov/DeepFaceLive synced 2025-08-20 05:23:23 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix AveCL

Browse source
This commit is contained in:
iperov 2021-10-01 18:10:06 +04:00
commit 8f530cc0a7
4 changed files with 55 additions and 25 deletions
Download patch file Download diff file Expand all files Collapse all files

10
xlib/avecl/_internal/HKernel.py
View file

@ -175,10 +175,7 @@ class HKernel:
line = f'#define {name_upper}_IDX({HKernel.axes_seq_enum(name, ndim)}) '
for i in range(ndim):
if i == 0:
line += f'( (size_t)({name_lower}{i}) )'
else:
line += f'( {name_lower}{i} )'
line += f'( (size_t)({name_lower}{i}) )'
for j in range(i+1,ndim):
line += f'*{shape[j]} '
@ -190,10 +187,7 @@ class HKernel:
line = f'#define {name_upper}_IDX_MOD({HKernel.axes_seq_enum(name, ndim)}) '
for i in range(ndim):
if i == 0:
line += f'( (size_t)({name_lower}{i}) % {shape[i]} )'
else:
line += f'( ({name_lower}{i}) % {shape[i]} )'
line += f'( (size_t)({name_lower}{i}) % {shape[i]} )'
for j in range(i+1,ndim):
line += f'*{shape[j]} '