Skip to main content

Browse the archive

Revision 68acb4b2d96dec56cf6f21a697f99ef52d241a64 authored by Jean Kossaifi on 11 February 2022, 19:15:52 UTC, committed by GitHub on 11 February 2022, 19:15:52 UTC 
Merge pull request #356 from MarieRoald/pytorch_sum_keepdims
 
Add `keepdims` to `tl.sum` with the PyTorch backend
2 parent s 97702a7 + 0b01b06
Raw File
Makefile 
# Automate testing etc
BACKEND?='numpy'

.PHONY: all test install clean debug

all: install test

install:
	pip install -e .

debug:
	TENSORLY_BACKEND=$(BACKEND) pytest -v --pdb tensorly

test:
	TENSORLY_BACKEND=$(BACKEND) pytest -v tensorly

test-all:
	TENSORLY_BACKEND='numpy' pytest -v tensorly
	TENSORLY_BACKEND='cupy' pytest -v tensorly
	TENSORLY_BACKEND='pytorch' pytest -v tensorly
	TENSORLY_BACKEND='mxnet' pytest -v tensorly
	TENSORLY_BACKEND='jax' pytest -v tensorly
	TENSORLY_BACKEND='tensorflow' pytest -v tensorly

test-coverage:
	TENSORLY_BACKEND=$(BACKEND) pytest -v --cov tensorly tensorly
back to top