fix flake8

cupy · Nov 19, 2018 · 754f368 · 754f368
1 parent ed10e3b
commit 754f368
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/cupy/cuda/cudnn.pyx b/cupy/cuda/cudnn.pyx
@@ -450,10 +450,10 @@ cdef extern from "cupy_cudnn.h" nogil:
     # CTC
     int cudnnCreateCTCLossDescriptor(CTCLossDescriptor* ctcLossDesc)
     int cudnnDestroyCTCLossDescriptor(CTCLossDescriptor ctcLossDesc)
-    int cudnnSetCTCLossDescriptor(CTCLossDescriptor ctcLossDesc,
-        DataType dataType)
-    int cudnnGetCTCLossDescriptor(CTCLossDescriptor ctcLossDesc,
-        DataType* dataType)
+    int cudnnSetCTCLossDescriptor(
+        CTCLossDescriptor ctcLossDesc, DataType dataType)
+    int cudnnGetCTCLossDescriptor(
+        CTCLossDescriptor ctcLossDesc, DataType* dataType)
     int cudnnGetCTCLossWorkspaceSize(
         Handle handle, TensorDescriptor probsDesc,
         TensorDescriptor gradientsDesc, int* labels,
@@ -1724,8 +1724,8 @@ cpdef setCTCLossDescriptor(size_t ctcLossDesc, int dataType):
 
 cpdef getCTCLossDescriptor(size_t ctcLossDesc):
     cdef DataType compType
-    status = cudnnGetCTCLossDescriptor(<CTCLossDescriptor>ctcLossDesc,
-        &compType)
+    status = cudnnGetCTCLossDescriptor(
+        <CTCLossDescriptor>ctcLossDesc, &compType)
     check_status(status)
     return compType