Exemplo n.º 1
0
    def cuda_index(self, input_points, extra_data=None):

        from cuda_indexing import CudaIndexing

        self.cuda_indexing = CudaIndexing()

        # we have only one table
        indexed_data = self.cuda_indexing.batch_indexing(
            self.uniform_planes[0], input_points)

        #for data in indexed_data:
        #    self.hash_tables[0].append_val(data, extra_data)
        #    extra_data += 1

        self.hash_tables[0].batch_append_vals(indexed_data.tolist(),
                                              extra_data)