Revision 9a94756d01d9071ff1610bfc4cb170bd47f701a8 authored by Alexander Root on 21 July 2022, 15:01:16 UTC, committed by GitHub on 21 July 2022, 15:01:16 UTC
* use pmaddubsw 8-bit horizontal widening adds * add SSE3 versions too * add pmaddubsw tests
1 parent 967c3bf
PythonExtensionGen.h
#ifndef HALIDE_PYTHON_EXTENSION_GEN_H_
#define HALIDE_PYTHON_EXTENSION_GEN_H_
#include <iostream>
#include <string>
#include <vector>
namespace Halide {
class Module;
namespace Internal {
struct LoweredArgument;
struct LoweredFunc;
class PythonExtensionGen {
public:
PythonExtensionGen(std::ostream &dest);
void compile(const Module &module);
private:
std::ostream &dest;
std::vector<std::string> buffer_refs;
void compile(const LoweredFunc &f);
void convert_buffer(const std::string &name, const LoweredArgument *arg);
void release_buffers(const std::string &prefix);
};
} // namespace Internal
} // namespace Halide
#endif // HALIDE_PYTHON_EXTENSION_GEN_H_
Computing file changes ...
