Skip to content
This repository was archived by the owner on Oct 15, 2019. It is now read-only.
This repository was archived by the owner on Oct 15, 2019. It is now read-only.

consider using THC #129

@soumith

Description

@soumith

Hi there devs, very good work!

I see that a lot of numpy functions are yet to be implemented for the GPU.
Instead of writing Tensor Op CUDA kernels from scratch, I suggest that you take a look at https://github.com/pytorch/pytorch/tree/master/torch/lib/THC
It is a C99 library with a lot of tensor functions implemented and might be helpful in saving time and effort.

This library currently powers torch and pytorch and has been developed and optimized for years.
The documentation for the functions is here: http://pytorch.org/docs/torch.html

Just a suggestion that might help accelerate your progress :)

cc: @tqchen

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions