File tree Expand file tree Collapse file tree 2 files changed +18
-6
lines changed
Expand file tree Collapse file tree 2 files changed +18
-6
lines changed Original file line number Diff line number Diff line change 8888 len* : cint # Number of elements allocated in memory
8989
9090
91- proc `=destroy` * (p: CudaLayoutArrayObj ) {.noSideEffect .}=
92- if not p.value.isNil:
93- discard cudaFree (p.value)
91+ when NimMajor == 1 :
92+ proc `=destroy` * (p: var CudaLayoutArrayObj ) {.noSideEffect .}=
93+ if not p.value.isNil:
94+ discard cudaFree (p.value)
95+ else :
96+ proc `=destroy` * (p: CudaLayoutArrayObj ) {.noSideEffect .}=
97+ if not p.value.isNil:
98+ discard cudaFree (p.value)
99+
100+
94101
95102proc layoutOnDevice * [T:SomeFloat ](t: CudaTensor [T]): CudaTensorLayout [T] {.noSideEffect .}=
96103 # # Store a CudaTensor shape, strides, etc information on the GPU
Original file line number Diff line number Diff line change 7979
8080 AnyTensor * [T] = Tensor [T] or CudaTensor [T] or ClTensor [T]
8181
82- proc `=destroy` * [T](p: CudaTensorRefTrackerObj [T]) {.noSideEffect .}=
83- if not p.value.isNil:
84- discard cudaFree (p.value)
82+ when NimMajor == 1 :
83+ proc `=destroy` * [T](p: var CudaTensorRefTrackerObj [T]) {.noSideEffect .}=
84+ if not p.value.isNil:
85+ discard cudaFree (p.value)
86+ else :
87+ proc `=destroy` * [T](p: CudaTensorRefTrackerObj [T]) {.noSideEffect .}=
88+ if not p.value.isNil:
89+ discard cudaFree (p.value)
8590
8691
8792
You can’t perform that action at this time.
0 commit comments