Git Product home page Git Product logo

sputnik's Introduction

Google Research

This repository contains code released by Google Research.

All datasets in this repository are released under the CC BY 4.0 International license, which can be found here: https://creativecommons.org/licenses/by/4.0/legalcode. All source files in this repository are released under the Apache 2.0 license, the text of which can be found in the LICENSE file.


Because the repo is large, we recommend you download only the subdirectory of interest:

  • Use GitHub editor to open the project. To open the editor change the url from github.com to github.dev in the address bar.
  • In the left navigation panel, right-click on the folder of interest and select download.

If you'd like to submit a pull request, you'll need to clone the repository; we recommend making a shallow clone (without history).

git clone [email protected]:google-research/google-research.git --depth=1

Disclaimer: This is not an official Google product.

Updated in 2023.

sputnik's People

Contributors

chr1sj0nes avatar tgale96 avatar

Stargazers

 avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar

Watchers

 avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar

sputnik's Issues

How to run SpMM with half precision?

How to run SpMM with half precision? I see that Sputnik only provides an interface for half2. If I have a CSR matrix with val in half precision, do I need to preprocess it? Also, why is the type of column idx is short2? If the matrix is very large, won't it be impossible to store the column indices?

cudaError_t CudaSpmm(int m, int k, int n, int nonzeros,
                     const int* __restrict__ row_indices,
                     const half2* __restrict__ values,
                     const int* __restrict__ row_offsets,
                     const short2* __restrict__ column_indices,
                     const half2* __restrict__ dense_matrix,
                     half2* __restrict__ output_matrix, cudaStream_t stream)

Any response would be helpful, thank you.

Run test without googletest dependencies

I want to run the code for a single matrix without depending on googletest, e.g., something like "/.spmm [options] tmp.mtx". Is that possible with the current code? P.S. the code runs and passes all the test cases performed by spmm_test.

efficient way to convert SDDMM output to dense

Thanks for releasing the code.
In your test API, I saw that you are using the Scipy api to convert the sparse output to the dense. However, I guess this is not an efficient way to convert it back. I would like to know, do you have any suggestion to make it more efficient? More precisely, I would like to know how I can get a dense output for SDDMM without sacrificing the performance?

SPMM

When initializing the sparse_tile_loader,the threadIdx.x should be threadIdx.x%kBlockWidth. Is what I said correct ?

No user API to create a SparseMatrix object from float* A

It seems that No user API is provided in your code to create a "SparseMatrix" object from a sparse matrix with a dense format (such as float* A).
Otherwise, I can not test the performance of your kernel with real sparse weight matrices.
I have to write my own codes to generate the CSR format from float* A, and then I have to generate the row indexes according to your row-swizzle policy.

Is my understanding correct or did I miss something in your code?

Recommend Projects

  • React photo React

    A declarative, efficient, and flexible JavaScript library for building user interfaces.

  • Vue.js photo Vue.js

    🖖 Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.

  • Typescript photo Typescript

    TypeScript is a superset of JavaScript that compiles to clean JavaScript output.

  • TensorFlow photo TensorFlow

    An Open Source Machine Learning Framework for Everyone

  • Django photo Django

    The Web framework for perfectionists with deadlines.

  • D3 photo D3

    Bring data to life with SVG, Canvas and HTML. 📊📈🎉

Recommend Topics

  • javascript

    JavaScript (JS) is a lightweight interpreted programming language with first-class functions.

  • web

    Some thing interesting about web. New door for the world.

  • server

    A server is a program made to process requests and deliver data to clients.

  • Machine learning

    Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.

  • Game

    Some thing interesting about game, make everyone happy.

Recommend Org

  • Facebook photo Facebook

    We are working to build community through open source technology. NB: members must have two-factor auth.

  • Microsoft photo Microsoft

    Open source projects and samples from Microsoft.

  • Google photo Google

    Google ❤️ Open Source for everyone.

  • D3 photo D3

    Data-Driven Documents codes.