Ai
69 Star 305 Fork 197

MindSpore/akg
暂停

加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
文件
克隆/下载
pack_func_convert.go 1.71 KB
一键复制 编辑 原始数据 按行查看 历史
xsgongxiaoqing 提交于 2021-03-18 21:30 +08:00 . init
/*
* Licensed to the Apache Software Foundation (ASF) under one
* or more contributor license agreements. See the NOTICE file
* distributed with this work for additional information
* regarding copyright ownership. The ASF licenses this file
* to you under the Apache License, Version 2.0 (the
* "License"); you may not use this file except in compliance
* with the License. You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
* KIND, either express or implied. See the License for the
* specific language governing permissions and limitations
* under the License.
*/
/*!
* \brief Sample golang application to demonstrate function conversion to packed function.
* \file pack_func_convert.go
*/
package main
import (
"fmt"
"./gotvm"
)
// sampleCb is a simple golang callback function like C = A + B.
func sampleCb(args ...*gotvm.Value) (retVal interface{}, err error) {
for _, v := range args {
fmt.Printf("ARGS:%T : %v\n", v.AsInt64(), v.AsInt64())
}
val1 := args[0].AsInt64()
val2 := args[1].AsInt64()
retVal = int64(val1+val2)
return
}
// main
func main() {
// Welcome
// Simple convert to a packed function
fhandle, err := gotvm.ConvertFunction(sampleCb)
if err != nil {
fmt.Print(err)
return
}
fmt.Printf("Converted function\n")
retVal, err := fhandle.Invoke(10, 20)
fmt.Printf("Invoke Completed\n")
if err != nil {
fmt.Print(err)
return
}
fmt.Printf("Result:%v\n", retVal.AsInt64())
}
Loading...
马建仓 AI 助手
尝试更多
代码解读
代码找茬
代码优化
Python
1
https://gitee.com/mindspore/akg.git
git@gitee.com:mindspore/akg.git
mindspore
akg
akg
v2.1.0

搜索帮助