drbh

feat: initial port of megablocks to builder format

2595c46 over 1 year ago

336 Bytes

#pragma once

#include <torch/all.h>

namespace megablocks {

// Public interface function for constructing indices from padded bins
void indices(torch::Tensor padded_bins,
             int block_size,
             int output_block_rows,
             int output_block_columns,
             torch::Tensor out);

} // namespace megablocks

	#pragma once

	#include <torch/all.h>

	namespace megablocks {

	// Public interface function for constructing indices from padded bins
	void indices(torch::Tensor padded_bins,
	int block_size,
	int output_block_rows,
	int output_block_columns,
	torch::Tensor out);

	} // namespace megablocks