FasTensor  1.0.0
Transform Supercomputing for AI
Macros | Functions
ft_example_vector_2d.cpp File Reference
#include <iostream>
#include <stdarg.h>
#include <vector>
#include <stdlib.h>
#include "ft.h"

Macros

#define INPUT_CHK_ROWS   4
 
#define INPUT_CHK_COLS   4
 
#define OUTPUT_VEC_ROWS   2
 
#define OUTPUT_VEC_COLS   2
 

Functions

Stencil< std::vector< float > > udf_vector_2d (const Stencil< float > &iStencil)
 UDF to support output of 2D vector. More...
 
int main (int argc, char *argv[])
 

Macro Definition Documentation

◆ INPUT_CHK_COLS

#define INPUT_CHK_COLS   4

◆ INPUT_CHK_ROWS

#define INPUT_CHK_ROWS   4

◆ OUTPUT_VEC_COLS

#define OUTPUT_VEC_COLS   2

◆ OUTPUT_VEC_ROWS

#define OUTPUT_VEC_ROWS   2

Function Documentation

◆ main()

int main ( int  argc,
char *  argv[] 
)

◆ udf_vector_2d()

Stencil<std::vector<float> > udf_vector_2d ( const Stencil< float > &  iStencil)
inline

UDF to support output of 2D vector.

Parameters
iStencil: input stencil
Returns
Stencil<std::vector<float>> : output stencil
Examples
ft_example_vector_2d.cpp.