Use TVM for TI DSP

jianjun · November 23, 2018, 11:56pm

Hello geeks, first time here.
I am learning TVM now and I am working on a TI DSP with two c66x cores.I have seen that TVM can generate souce code for CUDA/OpenCL, so I am wondering if I could extend this to generate optimized c66x souce code, which use “#Pragma roll/iterate before the for loop” and “restrict keyword before the variable” to accelerate the loops. Would you give me some suggestions?

merrymercy · November 25, 2018, 9:04am

You can start by looking the codegen for some backends, such as opencl (already uses _restrict), HLS (uses pragma).

github.com

dmlc/tvm/blob/master/src/codegen/codegen_opencl.cc

/*!
 *  Copyright (c) 2017 by Contributors
 * \file codegen_opencl.cc
 */
#include <tvm/packed_func_ext.h>
#include <vector>
#include <string>
#include "codegen_opencl.h"
#include "build_common.h"
#include "../runtime/thread_storage_scope.h"
#include "../runtime/opencl/opencl_module.h"

namespace tvm {
namespace codegen {

CodeGenOpenCL::CodeGenOpenCL() {
  restrict_keyword_ = "restrict";
}

void CodeGenOpenCL::InitFuncState(LoweredFunc f) {

This file has been truncated. show original

github.com

dmlc/tvm/blob/master/src/codegen/codegen_vhls.cc

/*!
 *  Copyright (c) 2018 by Contributors
 * \file codegen_vhls.cc
 */
#include <tvm/build_module.h>
#include <vector>
#include <string>
#include "codegen_vhls.h"
#include "build_common.h"
#include "../runtime/opencl/sdaccel/sdaccel_module.h"

namespace tvm {
namespace codegen {

void CodeGenVivadoHLS::Init(bool output_ssa) {
  CodeGenC::Init(output_ssa);

  this->stream << "#include <ap_int.h>\n\n";
  this->stream << "#include <algorithm>\n\n";
}

This file has been truncated. show original

jianjun · November 26, 2018, 12:37am

Thank you for your reply.

I will try to implement it.

briantse · May 6, 2019, 1:14am

@jianjun, hi, TVM supports the TI DSP such as TDA2X?

LanTn · May 8, 2019, 1:15am

hi,I am using TVM to support a DSP similar to TI architecture，but i dont know how to implement it；so I would like to ask you something about it and ask for help，if you have time ，please reply me，Thanks

jianjun · May 21, 2019, 2:12pm

I don’t think so, you have to extend by yourself.

jianjun · May 22, 2019, 2:01am

I didn’t implement one neither.

tico · January 20, 2020, 3:10pm

Hi,

Is there any update on support for TI DSPs or guidelines to extend TVM code gen for new targets?

Thanks