1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
|
package tensor
import (
"github.com/pkg/errors"
"gorgonia.org/tensor/internal/execution"
)
// StdEng is the default execution engine that comes with the tensors. To use other execution engines, use the WithEngine construction option.
type StdEng struct {
execution.E
}
// makeArray allocates a slice for the array
func (e StdEng) makeArray(arr *array, t Dtype, size int) {
arr.Raw = malloc(t, size)
arr.t = t
}
func (e StdEng) AllocAccessible() bool { return true }
func (e StdEng) Alloc(size int64) (Memory, error) { return nil, noopError{} }
func (e StdEng) Free(mem Memory, size int64) error { return nil }
func (e StdEng) Memset(mem Memory, val interface{}) error {
if ms, ok := mem.(MemSetter); ok {
return ms.Memset(val)
}
return errors.Errorf("Cannot memset %v with StdEng", mem)
}
func (e StdEng) Memclr(mem Memory) {
if z, ok := mem.(Zeroer); ok {
z.Zero()
}
return
}
func (e StdEng) Memcpy(dst, src Memory) error {
switch dt := dst.(type) {
case *array:
switch st := src.(type) {
case *array:
copyArray(dt, st)
return nil
case arrayer:
copyArray(dt, st.arrPtr())
return nil
}
case arrayer:
switch st := src.(type) {
case *array:
copyArray(dt.arrPtr(), st)
return nil
case arrayer:
copyArray(dt.arrPtr(), st.arrPtr())
return nil
}
}
return errors.Errorf("Failed to copy %T %T", dst, src)
}
func (e StdEng) Accessible(mem Memory) (Memory, error) { return mem, nil }
func (e StdEng) WorksWith(order DataOrder) bool { return true }
func (e StdEng) checkAccessible(t Tensor) error {
if !t.IsNativelyAccessible() {
return errors.Errorf(inaccessibleData, t)
}
return nil
}
|
