aboutsummaryrefslogtreecommitdiff
path: root/androidnn.go
blob: 92b7b2aa96d4d1a195a3be235223381bc8147917 (plain)
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
//
// Copyright © 2017 ARM Ltd. All rights reserved.
// SPDX-License-Identifier: MIT
//

package armnn_nn_driver

import (
    "android/soong/android"
    "android/soong/cc"
)

func globalFlags(ctx android.BaseContext) []string {
    var cppflags []string

    if ctx.AConfig().PlatformVersionName() == "Q" || ctx.AConfig().PlatformVersionName() == "10" {
        cppflags = append(cppflags, "-fno-addrsig")
    }

    return cppflags
}

func armnnNNDriverDefaults(ctx android.LoadHookContext) {
        type props struct {
                Cppflags []string
        }

        p := &props{}
        p.Cppflags = globalFlags(ctx)

        ctx.AppendProperties(p)
}

func init() {

  android.RegisterModuleType("armnn_nn_driver_defaults", armnnNNDriverDefaultsFactory)
}

func armnnNNDriverDefaultsFactory() android.Module {

   module := cc.DefaultsFactory()
   android.AddLoadHook(module, armnnNNDriverDefaults)
   return module
}